Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_aggregate_ops.cc89 TensorShape output_tf_shape; in ComputeScalar() local
135 TensorShape output_tf_shape; in Compute() local
H A Dmkl_pooling_ops_common.h565 TensorShape output_tf_shape; in AllocateEmptyOutputTensor() local
662 TensorShape output_tf_shape; in AllocateOutputTensor() local
725 TensorShape output_tf_shape; in AllocateOutputTensor() local
H A Dmkl_lrn_op.cc206 TensorShape output_tf_shape; in AllocateOutputTensor() local
469 TensorShape output_tf_shape; in AllocateOutputTensor() local
H A Dmkl_dequantize_op.cc112 TensorShape output_tf_shape; in Compute() local
H A Dmkl_matmul_op_fused.cc154 TensorShape output_tf_shape({batch, channel}); in Compute() local
H A Dmkl_softmax_op.cc301 TensorShape output_tf_shape; // shape of output TF tensor. in Compute() local
H A Dmkl_slice_op.cc486 TensorShape output_tf_shape; in AllocateOutputTensor() local
H A Dmkl_quantize_op.cc415 TensorShape output_tf_shape; in Compute() local
H A Dmkl_conv_grad_input_ops.cc581 TensorShape output_tf_shape; in AllocateOutputTensor() local
H A Dmkl_conv_grad_filter_ops.cc709 TensorShape output_tf_shape({output_dims_mkl_order[MklDnnDims::Dim_H], in AllocateOutputTensor() local
H A Dmkl_matmul_ops_common.h477 TensorShape output_tf_shape; variable
H A Dmkl_conv_ops.cc1106 TensorShape output_tf_shape; in AllocateOutputTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dmkl_util.h638 TensorShape output_tf_shape = input_mkl_shape.GetTfShape(); in ConvertMklToTF() local