| /external/tensorflow/tensorflow/core/kernels/ |
| 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,
482 CHECK_NOTNULL(output_dims_mkl_order);
503 output_dims_mkl_order, pad_l, pad_r,
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)
[all...] |
| mkl_pooling_ops_common.h | 491 memory::dims* output_dims_mkl_order) {
494 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch,
500 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch,
556 const memory::dims output_dims_mkl_order,
562 output_tf_shape = MklDnnDimsToTFShape(output_dims_mkl_order);
643 const memory::dims output_dims_mkl_order,
652 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(),
653 output_dims_mkl_order, output_tf_format);
695 const memory::dims output_dims_mkl_order,
704 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size()
[all...] |
| mkl_maxpooling_op.cc | 536 memory::dims output_dims_mkl_order; variable
537 this->GetOutputDims(pool_params, &output_dims_mkl_order);
543 output_dims_mkl_order, &output_tensor);
581 MklPoolingParams fwdParams(src_dims, output_dims_mkl_order, filter_dims,
588 output_dims_mkl_order,
591 dnn_data_output.SetUsrMem(output_dims_mkl_order,
730 memory::dims output_dims_mkl_order; variable
731 this->GetOutputDims(pool_params, &output_dims_mkl_order);
734 orig_input_dims_mkl_order, output_dims_mkl_order, filter_dims,
[all...] |
| mkl_avgpooling_op.cc | 466 memory::dims output_dims_mkl_order; variable
467 this->GetOutputDims(pool_params, &output_dims_mkl_order);
473 output_dims_mkl_order, &output_tensor);
504 MklPoolingParams fwdParams(src_dims, output_dims_mkl_order, filter_dims,
512 output_dims_mkl_order,
618 memory::dims output_dims_mkl_order; variable
619 this->GetOutputDims(pool_params, &output_dims_mkl_order);
624 orig_input_dims_mkl_order, output_dims_mkl_order, filter_dims,
|
| mkl_lrn_op.cc | 197 const memory::dims output_dims_mkl_order,
208 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(),
209 output_dims_mkl_order, output_tf_format);
447 const memory::dims output_dims_mkl_order,
458 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(),
459 output_dims_mkl_order, output_tf_format);
|
| mkl_conv_grad_filter_ops.cc | 489 // Conv2D: output_dims_mkl_order is in OIHW format.
520 // Conv3D: output_dims_mkl_order is in OIDHW format.
667 const memory::dims& output_dims_mkl_order,
679 // output_dims_mkl_order is in OIHW format.
681 TensorShape output_tf_shape({output_dims_mkl_order[MklDnnDims::Dim_H],
682 output_dims_mkl_order[MklDnnDims::Dim_W],
683 output_dims_mkl_order[MklDnnDims::Dim_I],
684 output_dims_mkl_order[MklDnnDims::Dim_O]});
|
| mkl_conv_grad_input_ops.cc | 536 const memory::dims& output_dims_mkl_order,
548 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(),
549 output_dims_mkl_order, output_tf_format);
|
| mkl_conv_ops.cc | [all...] |
