Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_lrn_op.cc241 TensorShape workspace_tf_shape; in MklDefaultToEigen() local
273 TensorShape workspace_tf_shape; in AllocateWorkspaceTensor() local
Dmkl_maxpooling_op.cc212 TensorShape workspace_tf_shape; in AllocateWorkspaceTensor() local
Dmkl_fused_batch_norm_op.cc723 TensorShape workspace_tf_shape; in Compute() local
940 TensorShape workspace_tf_shape, in HandleEmptyInput()
965 TensorShape workspace_tf_shape, in AllocateTFOutputs()