Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_aggregate_ops.cc90 TensorShape output_tf_shape; in ComputeScalar() local
136 TensorShape output_tf_shape; in Compute() local
Dmkl_dequantize_op.cc94 TensorShape output_tf_shape; in Compute() local
Dmkl_matmul_op_fused.cc140 TensorShape output_tf_shape({batch, channel}); in Compute() local
Dmkl_pooling_ops_common.h550 TensorShape output_tf_shape; in AllocateEmptyOutputTensor() local
647 TensorShape output_tf_shape; in AllocateOutputTensor() local
710 TensorShape output_tf_shape; in AllocateOutputTensor() local
Dmkl_lrn_op.cc205 TensorShape output_tf_shape; in AllocateOutputTensor() local
468 TensorShape output_tf_shape; in AllocateOutputTensor() local
Dmkl_softmax_op.cc278 TensorShape output_tf_shape; // shape of output TF tensor. in Compute() local
Dmkl_slice_op.cc478 TensorShape output_tf_shape; in AllocateOutputTensor() local
Dmkl_quantize_op.cc406 TensorShape output_tf_shape; in Compute() local
Dmkl_conv_grad_input_ops.cc578 TensorShape output_tf_shape; in AllocateOutputTensor() local
Dmkl_conv_grad_filter_ops.cc689 TensorShape output_tf_shape({output_dims_mkl_order[MklDnnDims::Dim_H], in AllocateOutputTensor() local
Dmkl_matmul_ops_common.h404 TensorShape output_tf_shape; in AllocateOutputTensor() local
Dmkl_conv_ops.cc870 TensorShape output_tf_shape; in AllocateOutputTensor() local
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h635 TensorShape output_tf_shape = input_mkl_shape.GetTfShape(); in ConvertMklToTF() local