Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_lrn_op.cc242 TensorShape workspace_tf_shape; in MklDefaultToEigen() local
274 TensorShape workspace_tf_shape; in AllocateWorkspaceTensor() local
H A Dmkl_fused_batch_norm_op.cc784 TensorShape workspace_tf_shape; in Compute() local
1002 TensorShape workspace_tf_shape, in HandleEmptyInput()
1027 TensorShape workspace_tf_shape, in AllocateTFOutputs()
H A Dmkl_maxpooling_op.cc218 TensorShape workspace_tf_shape; in AllocateWorkspaceTensor() local