(base) tb@tbs-Mac-mini tensorflow_code % python tf_ex.py
Init Plugin
Init Graph Optimizer
Init Kernel
Metal device set to: Apple M1
2021-07-02 10:25:17.792259: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:305] Could not identify NUMA node of platform GPU ID 0, defaulting to 0. Your kernel may not have been built with NUMA support.
2021-07-02 10:25:17.792589: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:271] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 0 MB memory) -> physical PluggableDevice (device: 0, name: METAL, pci bus id: <undefined>)
Model: "sequential"
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
dense (Dense)                (None, 512)               401920    
_________________________________________________________________
dropout (Dropout)            (None, 512)               0         
_________________________________________________________________
dense_1 (Dense)              (None, 1000)              513000    
_________________________________________________________________
dense_2 (Dense)              (None, 10)                10010     
=================================================================
Total params: 924,930
Trainable params: 924,930
Non-trainable params: 0
_________________________________________________________________
2021-07-02 10:25:17.852648: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:176] None of the MLIR Optimization Passes are enabled (registered 2)
2021-07-02 10:25:17.852834: W tensorflow/core/platform/profile_utils/cpu_utils.cc:128] Failed to get CPU frequency: 0 Hz
Epoch 1/10
2021-07-02 10:25:17.957554: I tensorflow/core/grappler/optimizers/custom_graph_optimizer_registry.cc:112] Plugin optimizer for device_type GPU is enabled.
2021-07-02 10:25:18.010 python[1941:97893] -[MPSNDArrayIdentity encodeToCommandEncoder:commandBuffer:sourceArrays:resultState:destinationArray:kernelDAGObject:]: unrecognized selector sent to instance 0x137c3d170
2021-07-02 10:25:18.013 python[1941:97893] *** Terminating app due to uncaught exception 'NSInvalidArgumentException', reason: '-[MPSNDArrayIdentity encodeToCommandEncoder:commandBuffer:sourceArrays:resultState:destinationArray:kernelDAGObject:]: unrecognized selector sent to instance 0x137c3d170'
*** First throw call stack:
(
        0   CoreFoundation                      0x00000001963dd320 __exceptionPreprocess + 240
        1   libobjc.A.dylib                     0x000000019610bc04 objc_exception_throw + 60
        2   CoreFoundation                      0x000000019646c020 -[NSObject(NSObject) __retain_OA] + 0
        3   CoreFoundation                      0x000000019633f174 ___forwarding___ + 1444
        4   CoreFoundation                      0x000000019633eb20 _CF_forwarding_prep_0 + 96
        5   libmetal_plugin.dylib               0x0000000124161db4 ___ZN12metal_plugin14MPSApplyAdamOpIfE7ComputeEPNS_15OpKernelContextE_block_invoke.115 + 92
        6   libdispatch.dylib                   0x00000001960b8420 _dispatch_client_callout + 20
        7   libdispatch.dylib                   0x00000001960c6a98 _dispatch_lane_barrier_sync_invoke_and_complete + 60
        8   libmetal_plugin.dylib               0x000000012415f3c0 _ZN12metal_plugin14MPSApplyAdamOpIfE7ComputeEPNS_15OpKernelContextE + 4884
        9   libmetal_plugin.dylib               0x000000012415de78 _ZN12metal_pluginL15ComputeOpKernelINS_14MPSApplyAdamOpIfEEEEvPvP18TF_OpKernelContext + 44
        10  _pywrap_tensorflow_internal.so      0x000000028007647c _ZN10tensorflow15PluggableDevice7ComputeEPNS_8OpKernelEPNS_15OpKernelContextE + 148
        11  libtensorflow_framework.2.dylib     0x0000000118560664 _ZN10tensorflow12_GLOBAL__N_113ExecutorStateINS_15PropagatorStateEE7ProcessENS2_10TaggedNodeEx + 3080
        12  libtensorflow_framework.2.dylib     0x0000000118561dbc _ZNSt3__110__function6__funcIZN10tensorflow12_GLOBAL__N_113ExecutorStateINS2_15PropagatorStateEE7RunTaskIZNS6_13ScheduleReadyEPN4absl14lts_2020_09_2313InlinedVectorINS5_10TaggedNodeELm8ENS_9allocatorISB_EEEEPNS5_20TaggedNodeReadyQueueEEUlvE0_EEvOT_EUlvE_NSC_ISL_EEFvvEEclEv + 56
        13  _pywrap_tensorflow_internal.so      0x00000002807aa228 _ZN5Eigen15ThreadPoolTemplIN10tensorflow6thread16EigenEnvironmentEE10WorkerLoopEi + 1508
        14  _pywrap_tensorflow_internal.so      0x00000002807a9b20 _ZZN10tensorflow6thread16EigenEnvironment12CreateThreadENSt3__18functionIFvvEEEENKUlvE_clEv + 80
        15  libtensorflow_framework.2.dylib     0x0000000118afcb08 _ZN10tensorflow12_GLOBAL__N_17PThread8ThreadFnEPv + 120
        16  libsystem_pthread.dylib             0x000000019626506c _pthread_start + 320
        17  libsystem_pthread.dylib             0x000000019625fda0 thread_start + 8
)
libc++abi.dylib: terminating with uncaught exception of type NSException
zsh: abort      python tf_ex.py