Home
last modified time | relevance | path

Searched refs:output_tf_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_softmax_op.cc138 TensorShape output_tf_shape; // shape of output TF tensor. in Compute() local
150 output_tf_shape.AddDim((dst_pd.get_size() / sizeof(T))); in Compute()
153 output_tf_shape = MklDnnDimsToTFShape(output_dims); in Compute()
156 AllocateOutputSetMklShape(context, 0, &output_tensor, output_tf_shape, in Compute()
Dmkl_pooling_ops_common.h560 TensorShape output_tf_shape; in AllocateEmptyOutputTensor() local
562 output_tf_shape = MklDnnDimsToTFShape(output_dims_mkl_order); in AllocateEmptyOutputTensor()
569 output_tf_shape = MklDnnDimsToTFShape(output_dims_NHWC_order); in AllocateEmptyOutputTensor()
572 output_tf_shape, output_mkl_shape); in AllocateEmptyOutputTensor()
654 TensorShape output_tf_shape; in AllocateOutputTensor() local
657 output_tf_shape.AddDim(this->GetNumTElements(dst_pd)); in AllocateOutputTensor()
659 output_tf_shape, output_mkl_shape); in AllocateOutputTensor()
707 TensorShape output_tf_shape; in AllocateOutputTensor() local
708 output_tf_shape.AddDim(this->GetNumTElements(dst_pd)); in AllocateOutputTensor()
710 output_tf_shape, output_mkl_shape); in AllocateOutputTensor()
Dmkl_aggregate_ops.cc215 TensorShape output_tf_shape; in Compute() local
231 output_tf_shape.AddDim((output_pd.get_size() / sizeof(T))); in Compute()
234 output_tf_shape = src1_tensor.shape(); in Compute()
236 AllocateOutputSetMklShape(ctx, output_idx, &dst_tensor, output_tf_shape, in Compute()
Dmkl_lrn_op.cc210 TensorShape output_tf_shape; in AllocateOutputTensor() local
214 output_tf_shape.AddDim(num_bytes / sizeof(T)); in AllocateOutputTensor()
216 output_tf_shape, output_mkl_shape); in AllocateOutputTensor()
461 TensorShape output_tf_shape; in AllocateOutputTensor() local
464 output_tf_shape.AddDim(num_bytes / sizeof(T)); in AllocateOutputTensor()
466 output_tf_shape, output_mkl_shape); in AllocateOutputTensor()
Dmkl_slice_op.cc452 TensorShape output_tf_shape; in AllocateOutputTensor() local
465 output_tf_shape.AddDim(output_pd->get_size() / sizeof(T)); in AllocateOutputTensor()
469 output_tf_shape = MklDnnDimsToTFShape(output_dims); in AllocateOutputTensor()
472 AllocateOutputSetMklShape(context, 0, output_tensor, output_tf_shape, in AllocateOutputTensor()
Dmkl_conv_grad_input_ops.cc552 TensorShape output_tf_shape; in AllocateOutputTensor() local
553 output_tf_shape.AddDim(dst_pd.get_size() / sizeof(T)); in AllocateOutputTensor()
555 AllocateOutputSetMklShape(context, 0, output_tensor, output_tf_shape, in AllocateOutputTensor()
Dmkl_conv_grad_filter_ops.cc681 TensorShape output_tf_shape({output_dims_mkl_order[MklDnnDims::Dim_H], in AllocateOutputTensor() local
685 AllocateOutputSetMklShape(context, 0, output_tensor, output_tf_shape, in AllocateOutputTensor()
Dmkl_conv_ops.cc1272 TensorShape output_tf_shape; in AllocateOutputTensor() local
1273 output_tf_shape.AddDim((dst_pd.get_size() / sizeof(Toutput))); in AllocateOutputTensor()
1276 output_tf_shape, output_mkl_shape); in AllocateOutputTensor()