Home
last modified time | relevance | path

Searched defs:tf_shape_src (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_fused_batch_norm_op.cc715 TensorShape tf_shape_src; in Compute() local
1001 void HandleEmptyInput(OpKernelContext* context, TensorShape tf_shape_src, in HandleEmptyInput()
1146 TensorShape tf_shape_src, tf_shape_diff_dst; in Compute() local
1409 void HandleEmptyInput(OpKernelContext* context, TensorShape tf_shape_src, in HandleEmptyInput()