Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_avgpooling_op.cc73 memory::dims output_dims_mkl_order; in Compute() local
74 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
80 output_dims_mkl_order, &output_tensor); in Compute()
114 src_dims, output_dims_mkl_order, filter_dims, strides, padding_left, in Compute()
123 output_dims_mkl_order, in Compute()
223 memory::dims output_dims_mkl_order; in Compute() local
224 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
243 orig_input_dims_mkl_order, output_dims_mkl_order, filter_dims, in Compute()
Dmkl_maxpooling_op.cc81 memory::dims output_dims_mkl_order; in Compute() local
82 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
88 output_dims_mkl_order, &output_tensor); in Compute()
99 &pool_params, output_dims_mkl_order, in Compute()
140 src_dims, output_dims_mkl_order, filter_dims, strides, padding_left, in Compute()
147 output_dims_mkl_order, in Compute()
280 memory::dims output_dims_mkl_order; in Compute() local
281 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
298 orig_input_dims_mkl_order, output_dims_mkl_order, filter_dims, in Compute()
Dmkl_conv_ops.h321 memory::dims* output_dims_tf_order, memory::dims* output_dims_mkl_order,
325 DCHECK(output_dims_mkl_order);
469 *output_dims_mkl_order = mkldnn_sizes;
477 *output_dims_mkl_order = mkldnn_sizes;
488 memory::dims* output_dims_mkl_order, memory::dims* pad_l, in GetOutputAndPadSizeInMklOrder() argument
491 DCHECK(output_dims_mkl_order); in GetOutputAndPadSizeInMklOrder()
512 output_dims_mkl_order, pad_l, pad_r, in GetOutputAndPadSizeInMklOrder()
528 memory::dims* output_dims_tf_order, memory::dims* output_dims_mkl_order,
536 DCHECK(output_dims_mkl_order);
549 output_dims_mkl_order, pad_l, pad_r, pad_enabled, is_depthwise);
Dmkl_pooling_ops_common.h481 memory::dims* output_dims_mkl_order) { in GetOutputDims() argument
484 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch, in GetOutputDims()
490 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch, in GetOutputDims()
546 const memory::dims output_dims_mkl_order, in AllocateEmptyOutputTensor() argument
552 output_tf_shape = MklDnnDimsToTFShape(output_dims_mkl_order); in AllocateEmptyOutputTensor()
644 const memory::dims output_dims_mkl_order, in AllocateOutputTensor() argument
655 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
656 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
698 const memory::dims output_dims_mkl_order, in AllocateOutputTensor() argument
707 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
[all …]
Dmkl_lrn_op.cc192 const memory::dims output_dims_mkl_order, in AllocateOutputTensor() argument
203 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
204 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
455 const memory::dims output_dims_mkl_order, in AllocateOutputTensor() argument
465 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
466 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
Dmkl_conv_grad_filter_ops.cc675 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor() argument
689 TensorShape output_tf_shape({output_dims_mkl_order[MklDnnDims::Dim_H], in AllocateOutputTensor()
690 output_dims_mkl_order[MklDnnDims::Dim_W], in AllocateOutputTensor()
691 output_dims_mkl_order[MklDnnDims::Dim_I], in AllocateOutputTensor()
692 output_dims_mkl_order[MklDnnDims::Dim_O]}); in AllocateOutputTensor()
Dmkl_conv_grad_input_ops.cc561 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor() argument
574 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
575 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
Dmkl_matmul_ops_common.h392 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor() argument
401 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
402 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
Dmkl_conv_ops.cc850 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor() argument
866 output_mkl_shape->SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
867 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
905 : memory::desc(output_dims_mkl_order, MklDnnType<Toutput>(), in AllocateOutputTensor()
1740 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor() argument
1777 output_dims_mkl_order, in AllocateOutputTensor()
1819 : memory::desc(output_dims_mkl_order, MklDnnType<Tbias>(), in AllocateOutputTensor()