Home
last modified time | relevance | path

Searched refs:output_mkl_shape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_softmax_op.cc137 MklDnnShape output_mkl_shape; in Compute() local
145 output_mkl_shape.SetMklTensor(true); in Compute()
146 output_mkl_shape.SetMklLayout(&dst_pd); in Compute()
147 output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
148 output_mkl_shape.SetTfLayout(output_dims.size(), output_dims, in Compute()
152 output_mkl_shape.SetMklTensor(false); in Compute()
157 output_mkl_shape); in Compute()
Dmkl_aggregate_ops.cc214 MklDnnShape output_mkl_shape; in Compute() local
218 output_mkl_shape.SetMklTensor(true); in Compute()
220 output_mkl_shape.SetMklLayout(&output_pd); in Compute()
221 output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
223 output_mkl_shape.SetTfLayout(src1_dims_size, in Compute()
227 output_mkl_shape.SetTfLayout(src2_dims_size, in Compute()
233 output_mkl_shape.SetMklTensor(false); in Compute()
237 output_mkl_shape); in Compute()
Dmkl_slice_op.cc378 MklDnnShape output_mkl_shape; in ComputeMklSlice() local
385 output_mkl_shape); in ComputeMklSlice()
420 &output_tensor, &output_mkl_shape); in ComputeMklSlice()
422 DCHECK_EQ(input_mkl_shape.IsMklTensor(), output_mkl_shape.IsMklTensor()); in ComputeMklSlice()
448 MklDnnShape* output_mkl_shape) { in AllocateOutputTensor() argument
450 DCHECK(output_mkl_shape); in AllocateOutputTensor()
459 output_mkl_shape->SetMklTensor(true); in AllocateOutputTensor()
460 output_mkl_shape->SetMklLayout(output_pd); in AllocateOutputTensor()
461 output_mkl_shape->SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
462 output_mkl_shape->SetTfLayout(input_mkl_shape.GetDimension(), output_dims, in AllocateOutputTensor()
[all …]
Dmkl_pooling_ops_common.h558 MklDnnShape output_mkl_shape; in AllocateEmptyOutputTensor() local
559 output_mkl_shape.SetMklTensor(false); in AllocateEmptyOutputTensor()
572 output_tf_shape, output_mkl_shape); in AllocateEmptyOutputTensor()
648 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
649 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
650 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
651 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
652 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
659 output_tf_shape, output_mkl_shape); in AllocateOutputTensor()
700 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
[all …]
Dmkl_lrn_op.cc202 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
205 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
206 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
207 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
208 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
216 output_tf_shape, output_mkl_shape); in AllocateOutputTensor()
452 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
455 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
456 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
457 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
[all …]
Dmkl_conv_grad_input_ops.cc544 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
545 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
546 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
547 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
548 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
556 output_mkl_shape); in AllocateOutputTensor()
Dmkl_conv_grad_bias_ops.cc92 MklShape output_mkl_shape; in Compute() local
93 output_mkl_shape.SetMklTensor(false); in Compute()
95 output_mkl_shape); in Compute()
Dmkl_conv_grad_filter_ops.cc677 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
678 output_mkl_shape.SetMklTensor(false); in AllocateOutputTensor()
686 output_mkl_shape); in AllocateOutputTensor()
Dmkl_conv_ops.cc1264 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
1265 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
1266 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
1267 output_mkl_shape.SetElemType(MklDnnType<Toutput>()); in AllocateOutputTensor()
1268 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
1276 output_tf_shape, output_mkl_shape); in AllocateOutputTensor()