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.cc242 TensorShape workspace_tf_shape; in MklDefaultToEigen() local
274 TensorShape workspace_tf_shape; in AllocateWorkspaceTensor() local
Dmkl_maxpooling_op.cc213 TensorShape workspace_tf_shape; in AllocateWorkspaceTensor() local
Dmkl_fused_batch_norm_op.cc730 TensorShape workspace_tf_shape; in Compute() local
948 TensorShape workspace_tf_shape, in HandleEmptyInput()
973 TensorShape workspace_tf_shape, in AllocateTFOutputs()