| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_pooling_ops_common.h | 145 memory::dims* output_dims_mkl_order) {
147 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch,
216 const memory::dims output_dims_mkl_order,
225 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(),
226 output_dims_mkl_order, output_tf_format);
283 const memory::dims output_dims_mkl_order,
292 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(),
293 output_dims_mkl_order, output_tf_format);
354 memory::dims output_dims_mkl_order) {
355 this->GetOutputDims(pool_params, &output_dims_mkl_order);
[all...] |
| mkl_conv_ops.h | 217 memory::dims* output_dims_mkl_order, memory::dims* pad_l,
220 CHECK_NOTNULL(output_dims_mkl_order);
262 *output_dims_mkl_order = mkldnn_sizes;
275 memory::dims* output_dims_tf_order, memory::dims* output_dims_mkl_order,
278 CHECK_NOTNULL(output_dims_mkl_order);
290 output_dims_tf_order, output_dims_mkl_order,
304 memory::dims* output_dims_mkl_order, memory::dims* pad_l,
310 CHECK_NOTNULL(output_dims_mkl_order);
320 output_dims_tf_order, output_dims_mkl_order,
|
| mkl_avgpooling_op.cc | 465 memory::dims output_dims_mkl_order; variable
466 this->GetOutputDims(pool_params, &output_dims_mkl_order);
474 output_tf_shape = MklDnnDimsToTFShape(output_dims_mkl_order);
494 output_dims_mkl_order,
499 dnn_data_output.SetUsrMem(output_dims_mkl_order,
504 dnn_data_output.SetOpMemDesc(output_dims_mkl_order, memory::format::any);
520 this->AllocateOutputTensor(context, pool_prim_desc, output_dims_mkl_order,
575 memory::dims output_dims_mkl_order, original_input_dims_nchw; variable
585 pool_params, input_gradient_mkl_shape, output_dims_mkl_order);
|
| mkl_conv_ops.cc | 532 memory::dims output_dims_tf_order, output_dims_mkl_order; variable
540 &output_dims_tf_order, &output_dims_mkl_order, &padding_l,
590 output.SetUsrMem(output_dims_mkl_order, tf_fmt);
595 output.SetOpMemDesc(output_dims_mkl_order, memory::format::any);
614 AllocateOutputTensor(context, conv_prim_desc, output_dims_mkl_order,
635 AllocateOutputTensor(context, conv_prim_desc, output_dims_mkl_order,
668 const memory::dims& output_dims_mkl_order,
678 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(),
679 output_dims_mkl_order, output_tf_format);
[all...] |
| mkl_maxpooling_op.cc | 533 memory::dims output_dims_mkl_order; variable
534 this->GetOutputDims(pool_params, &output_dims_mkl_order);
540 output_dims_mkl_order,
544 dnn_data_output.SetUsrMem(output_dims_mkl_order,
549 dnn_data_output.SetOpMemDesc(output_dims_mkl_order, memory::format::any);
564 this->AllocateOutputTensor(context, pool_fwd_desc, output_dims_mkl_order,
649 memory::dims output_dims_mkl_order, orig_input_dims_mkl_order; variable
655 pool_params, orig_output_mkl_shape, output_dims_mkl_order);
[all...] |
| mkl_conv_grad_filter_ops.cc | 569 const memory::dims& output_dims_mkl_order,
581 // output_dims_mkl_order is in OIHW format.
583 TensorShape output_tf_shape({output_dims_mkl_order[MklDnnDims::Dim_H],
584 output_dims_mkl_order[MklDnnDims::Dim_W],
585 output_dims_mkl_order[MklDnnDims::Dim_I],
586 output_dims_mkl_order[MklDnnDims::Dim_O]});
|
| mkl_conv_grad_input_ops.cc | 456 const memory::dims& output_dims_mkl_order,
468 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(),
469 output_dims_mkl_order, output_tf_format);
|
| mkl_lrn_op.cc | [all...] |
