Home
last modified time | relevance | path

Searched refs: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
242 workspace_tf_shape.AddDim(0); in MklDefaultToEigen()
244 workspace_tf_shape, workspace_mkl_shape); in MklDefaultToEigen()
273 TensorShape workspace_tf_shape; in AllocateWorkspaceTensor() local
274 workspace_tf_shape.AddDim(workspace_bytes); in AllocateWorkspaceTensor()
276 workspace_tf_shape, workspace_mkl_shape); in AllocateWorkspaceTensor()
Dmkl_fused_batch_norm_op.cc723 TensorShape workspace_tf_shape; in Compute() local
726 workspace_tf_shape.AddDim(workspace_bytes); in Compute()
727 HandleEmptyInput(context, tf_shape_src, workspace_tf_shape, in Compute()
789 workspace_tf_shape.AddDim(workspace_bytes); in Compute()
791 AllocateTFOutputs(context, scale_tensor.shape(), workspace_tf_shape, in Compute()
802 workspace_tf_shape.AddDim(workspace_bytes); in Compute()
803 AllocateTFOutputs(context, scale_tensor.shape(), workspace_tf_shape, in Compute()
940 TensorShape workspace_tf_shape, in HandleEmptyInput() argument
958 AllocateTFOutputs(context, tf_shape_scale, workspace_tf_shape, in HandleEmptyInput()
965 TensorShape workspace_tf_shape, in AllocateTFOutputs() argument
[all …]
Dmkl_maxpooling_op.cc212 TensorShape workspace_tf_shape; in AllocateWorkspaceTensor() local
213 workspace_tf_shape.AddDim(workspace_bytes); in AllocateWorkspaceTensor()
215 &workspace_tensor, workspace_tf_shape, in AllocateWorkspaceTensor()