| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | 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);
144 GetOutputDims(const MklPoolParameters& mkl_pool_params, memory::dims* output_dims_mkl_order) argument
213 AllocateOutputTensor( OpKernelContext* context, const pooling_forward::primitive_desc& pool_fwd_prim_desc, const memory::dims output_dims_mkl_order, const memory::format& output_tf_format, Tensor** output_tensor) argument
280 AllocateOutputTensor( OpKernelContext* context, const pooling_backward::primitive_desc& pool_bkwd_prim_desc, const memory::dims output_dims_mkl_order, const memory::format& output_tf_format, Tensor** output_tensor) argument
351 ConfigureOriginalOutput( const MklPoolParameters& pool_params, const MklDnnShape& original_output_mkl_shape, memory::dims output_dims_mkl_order) argument
[all...] |
| H A D | 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,
214 GetOutputAndPadSizeInMklOrder( const TensorShape& input_shape, const TensorShape& filter_shape, const memory::dims& strides, memory::dims* output_dims_tf_order, memory::dims* output_dims_mkl_order, memory::dims* pad_l, memory::dims* pad_r) argument
273 GetOutputAndPadSizeInMklOrder( size_t src_index, size_t filter_index, const memory::dims& strides, memory::dims* output_dims_tf_order, memory::dims* output_dims_mkl_order, memory::dims* pad_l, memory::dims* pad_r) argument
300 GetConvFwdSizesInMklOrder( const TensorShape& input_shape, const TensorShape& filter_shape, memory::dims* input_dims, memory::dims* filter_dims, memory::dims* strides, memory::dims* output_dims_tf_order, memory::dims* output_dims_mkl_order, memory::dims* pad_l, memory::dims* pad_r) argument
|
| H A D | 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);
|
| H A D | 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);
665 AllocateOutputTensor( OpKernelContext* context, const convolution_forward::primitive_desc& conv_prim_desc, const memory::dims& output_dims_mkl_order, memory::format output_tf_format, Tensor** output_tensor) argument
|
| H A D | 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);
|
| H A D | 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]});
566 AllocateOutputTensor( OpKernelContext* context, const convolution_backward_weights::primitive_desc& conv_pd, const memory::dims& output_dims_mkl_order, memory::format output_tf_format, Tensor** output_tensor) argument
|
| H A D | 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);
453 AllocateOutputTensor( OpKernelContext* context, const convolution_backward_data::primitive_desc& conv_pd, const memory::dims& output_dims_mkl_order, memory::format output_tf_format, Tensor** output_tensor) argument
|
| H A D | mkl_lrn_op.cc | 868 const memory::dims output_dims_mkl_order,
879 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(),
880 output_dims_mkl_order, output_tf_format);
1118 const memory::dims output_dims_mkl_order,
1129 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(),
1130 output_dims_mkl_order, output_tf_format);
|