Home
last modified time | relevance | path

Searched refs:GetSizesAsMklDnnDims (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_input_conversion_op.cc128 input_shape_0.GetSizesAsMklDnnDims(), in Compute()
226 mkl_shape->GetSizesAsMklDnnDims(), in Compute()
Dmkl_aggregate_ops.cc224 src1_mkl_shape.GetSizesAsMklDnnDims(), in Compute()
228 src2_mkl_shape.GetSizesAsMklDnnDims(), in Compute()
Dmkl_relu_op.cc475 src_dims = dnn_shape_src.GetSizesAsMklDnnDims(); in Compute()
513 dnn_shape_src.GetSizesAsMklDnnDims(), in Compute()
597 src_dims = dnn_shape_src.GetSizesAsMklDnnDims(); in Compute()
625 src_dims = dnn_shape_src.GetSizesAsMklDnnDims(); in Compute()
676 dnn_shape_src.GetSizesAsMklDnnDims(), in Compute()
681 dnn_shape_diff_dst.GetSizesAsMklDnnDims(), in Compute()
Dmkl_avgpooling_op.cc485 ? dnn_shape_input.GetSizesAsMklDnnDims() in Compute()
605 ? orig_input_mkl_shape.GetSizesAsMklDnnDims() in Compute()
613 ? grad_mkl_shape.GetSizesAsMklDnnDims() in Compute()
Dmkl_softmax_op.cc68 output_dims = src_mkl_shape.GetSizesAsMklDnnDims(); in Compute()
Dmkl_lrn_op.cc127 memory::dims input_dims = src_dnn_shape.GetSizesAsMklDnnDims(); in Compute()
387 orig_input_dnn_shape.GetSizesAsMklDnnDims(); in Compute()
477 memory::dims orig_input_dims = input_grad_dnn_shape.GetSizesAsMklDnnDims(); in ConfigureInputGradient()
Dmkl_maxpooling_op.cc563 ? dnn_shape_input.GetSizesAsMklDnnDims() in Compute()
716 ? orig_input_mkl_shape.GetSizesAsMklDnnDims() in Compute()
724 ? grad_mkl_shape.GetSizesAsMklDnnDims() in Compute()
Dmkl_fused_batch_norm_op.cc612 ? dnn_shape_src.GetSizesAsMklDnnDims() in Compute()
926 ? dnn_shape_src.GetSizesAsMklDnnDims() in Compute()
930 ? dnn_shape_diff_dst.GetSizesAsMklDnnDims() in Compute()
Dmkl_pooling_ops_common.h744 ? input_gradient_mkl_shape.GetSizesAsMklDnnDims() in ConfigureInputGradient()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h567 inline memory::dims GetSizesAsMklDnnDims() const { in GetSizesAsMklDnnDims() function