Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_aggregate_ops.cc91 MklDnnShape output_mkl_shape; in ComputeScalar() local
101 output_mkl_shape.SetMklTensor(false); in ComputeScalar()
104 output_mkl_shape); in ComputeScalar()
137 MklDnnShape output_mkl_shape; in Compute() local
144 output_mkl_shape.SetMklTensor(false); in Compute()
147 output_mkl_shape); in Compute()
218 output_mkl_shape.SetMklTensor(has_mkl_input); in Compute()
223 output_mkl_shape.SetMklLayout(&output_pd); in Compute()
224 output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
225 output_mkl_shape.SetTfLayout(mkl_shape.GetDimension(), in Compute()
[all …]
Dmkl_dequantize_op.cc93 MklDnnShape output_mkl_shape; in Compute() local
109 output_mkl_shape.SetMklTensor(true); in Compute()
110 output_mkl_shape.SetMklLayout(&dst_md); in Compute()
111 output_mkl_shape.SetElemType(MklDnnType<float>()); in Compute()
112 output_mkl_shape.SetTfLayout(src_mkl_shape.GetDimension(), in Compute()
117 output_mkl_shape.SetMklTensor(false); in Compute()
123 output_mkl_shape); in Compute()
Dmkl_pooling_ops_common.h548 MklDnnShape output_mkl_shape; in AllocateEmptyOutputTensor() local
549 output_mkl_shape.SetMklTensor(false); in AllocateEmptyOutputTensor()
571 output_tf_shape, output_mkl_shape, in AllocateEmptyOutputTensor()
651 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
652 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
653 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
654 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
655 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
661 output_tf_shape = output_mkl_shape.GetTfShape(); in AllocateOutputTensor()
664 output_tf_shape, output_mkl_shape, in AllocateOutputTensor()
[all …]
Dmkl_slice_op.cc386 MklDnnShape output_mkl_shape; in ComputeMklSlice() local
393 output_mkl_shape); in ComputeMklSlice()
444 &output_tensor, &output_mkl_shape); in ComputeMklSlice()
446 DCHECK_EQ(input_mkl_shape.IsMklTensor(), output_mkl_shape.IsMklTensor()); in ComputeMklSlice()
474 MklDnnShape* output_mkl_shape) { in AllocateOutputTensor() argument
476 DCHECK(output_mkl_shape); in AllocateOutputTensor()
485 output_mkl_shape->SetMklTensor(true); in AllocateOutputTensor()
486 output_mkl_shape->SetMklLayout(output_pd); in AllocateOutputTensor()
487 output_mkl_shape->SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
488 output_mkl_shape->SetTfLayout(input_mkl_shape.GetDimension(), output_dims, in AllocateOutputTensor()
[all …]
Dmkl_quantize_op.cc269 MklDnnShape output_mkl_shape; in ComputeScalar() local
270 output_mkl_shape.SetMklTensor(false); in ComputeScalar()
274 output_mkl_shape); in ComputeScalar()
405 MklDnnShape output_mkl_shape; in Compute() local
408 output_mkl_shape.SetMklTensor(true); in Compute()
409 output_mkl_shape.SetMklLayout(&dst_md); in Compute()
410 output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
411 output_mkl_shape.SetTfLayout(src_mkl_shape.GetDimension(), in Compute()
416 output_mkl_shape.SetMklTensor(false); in Compute()
422 output_mkl_shape); in Compute()
Dmkl_softmax_op.cc277 MklDnnShape output_mkl_shape; in Compute() local
285 output_mkl_shape.SetMklTensor(true); in Compute()
286 output_mkl_shape.SetMklLayout(&dst_pd); in Compute()
287 output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
288 output_mkl_shape.SetTfLayout(src_dims.size(), src_dims, layout_type); in Compute()
291 output_mkl_shape.SetMklTensor(false); in Compute()
296 output_mkl_shape); in Compute()
Dmkl_lrn_op.cc197 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
200 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
201 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
202 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
203 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
211 output_tf_shape, output_mkl_shape); in AllocateOutputTensor()
459 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
462 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
463 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
464 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
[all …]
Dmkl_matmul_op_fused.cc137 MklDnnShape output_mkl_shape; in Compute() local
138 output_mkl_shape.SetMklTensor(false); in Compute()
154 &dst_tensor, output_mkl_shape, false)) { in Compute()
160 output_tf_shape, output_mkl_shape, in Compute()
193 output_mkl_shape, native_format); in Compute()
Dmkl_conv_grad_input_ops.cc570 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
571 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
572 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
573 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
574 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
582 output_mkl_shape); in AllocateOutputTensor()
Dmkl_matmul_ops_common.h397 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
398 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
399 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()
400 output_mkl_shape.SetElemType(MklDnnType<Toutput>()); in AllocateOutputTensor()
401 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
409 output_tf_shape, output_mkl_shape); in AllocateOutputTensor()
Dmkl_conv_ops.cc664 MklDnnShape output_mkl_shape; in Compute() local
667 &output_mkl_shape, &dst_tensor); in Compute()
852 MklDnnShape* output_mkl_shape, in AllocateOutputTensor() argument
863 output_mkl_shape->SetMklTensor(true); in AllocateOutputTensor()
864 output_mkl_shape->SetMklLayout(&dst_md); in AllocateOutputTensor()
865 output_mkl_shape->SetElemType(MklDnnType<Toutput>()); in AllocateOutputTensor()
866 output_mkl_shape->SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
873 output_tf_shape = output_mkl_shape->GetTfShape(); in AllocateOutputTensor()
888 if (!native_format && add_mkl_shape == *output_mkl_shape && in AllocateOutputTensor()
895 output_tf_shape, *output_mkl_shape, in AllocateOutputTensor()
[all …]
Dmkl_conv_grad_filter_ops.cc685 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
686 output_mkl_shape.SetMklTensor(false); in AllocateOutputTensor()
694 output_mkl_shape); in AllocateOutputTensor()