| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.h | 315 memory::dims* output_dims_tf_order, memory::dims* output_dims_mkl_order,
319 CHECK_NOTNULL(output_dims_mkl_order);
460 *output_dims_mkl_order = mkldnn_sizes;
468 *output_dims_mkl_order = mkldnn_sizes;
479 memory::dims* output_dims_mkl_order, memory::dims* pad_l, in GetOutputAndPadSizeInMklOrder() argument
482 CHECK_NOTNULL(output_dims_mkl_order); in GetOutputAndPadSizeInMklOrder()
503 output_dims_mkl_order, pad_l, pad_r, in GetOutputAndPadSizeInMklOrder()
519 memory::dims* output_dims_tf_order, memory::dims* output_dims_mkl_order,
527 CHECK_NOTNULL(output_dims_mkl_order);
540 output_dims_mkl_order, pad_l, pad_r, pad_enabled, is_depthwise);
|
| D | mkl_pooling_ops_common.h | 491 memory::dims* output_dims_mkl_order) { in GetOutputDims() argument
494 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch, in GetOutputDims()
500 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch, in GetOutputDims()
556 const memory::dims output_dims_mkl_order, in AllocateEmptyOutputTensor() argument
562 output_tf_shape = MklDnnDimsToTFShape(output_dims_mkl_order); in AllocateEmptyOutputTensor()
643 const memory::dims output_dims_mkl_order, in AllocateOutputTensor() argument
652 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
653 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
695 const memory::dims output_dims_mkl_order, in AllocateOutputTensor() argument
704 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
[all …]
|
| D | mkl_avgpooling_op.cc | 466 memory::dims output_dims_mkl_order; in Compute() local
467 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
473 output_dims_mkl_order, &output_tensor); in Compute()
504 MklPoolingParams fwdParams(src_dims, output_dims_mkl_order, filter_dims, in Compute()
512 output_dims_mkl_order, in Compute()
618 memory::dims output_dims_mkl_order; in Compute() local
619 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
624 orig_input_dims_mkl_order, output_dims_mkl_order, filter_dims, in Compute()
|
| D | mkl_maxpooling_op.cc | 536 memory::dims output_dims_mkl_order; in Compute() local
537 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
543 output_dims_mkl_order, &output_tensor); in Compute()
581 MklPoolingParams fwdParams(src_dims, output_dims_mkl_order, filter_dims, in Compute()
588 output_dims_mkl_order, in Compute()
591 dnn_data_output.SetUsrMem(output_dims_mkl_order, in Compute()
730 memory::dims output_dims_mkl_order; in Compute() local
731 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
734 orig_input_dims_mkl_order, output_dims_mkl_order, filter_dims, in Compute()
|
| D | mkl_lrn_op.cc | 197 const memory::dims output_dims_mkl_order, in AllocateOutputTensor() argument
208 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
209 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
447 const memory::dims output_dims_mkl_order, in AllocateOutputTensor() argument
458 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
459 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
|
| D | mkl_conv_grad_filter_ops.cc | 667 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor() argument
681 TensorShape output_tf_shape({output_dims_mkl_order[MklDnnDims::Dim_H], in AllocateOutputTensor()
682 output_dims_mkl_order[MklDnnDims::Dim_W], in AllocateOutputTensor()
683 output_dims_mkl_order[MklDnnDims::Dim_I], in AllocateOutputTensor()
684 output_dims_mkl_order[MklDnnDims::Dim_O]}); in AllocateOutputTensor()
|
| D | mkl_conv_grad_input_ops.cc | 536 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor() argument
548 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
549 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
|
| D | mkl_conv_ops.cc | 1251 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor() argument
1268 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
1269 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
1757 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor() argument
1794 output_dims_mkl_order, in AllocateOutputTensor()
1824 : memory::desc(output_dims_mkl_order, MklDnnType<Tbias>(), in AllocateOutputTensor()
|