Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_dequantize_op.cc69 ? src_mkl_shape.GetSizesAsMklDnnDims() in Compute()
113 src_mkl_shape.GetSizesAsMklDnnDims(), in Compute()
Dmkl_avgpooling_op.cc92 ? dnn_shape_input.GetSizesAsMklDnnDims() in Compute()
210 ? orig_input_mkl_shape.GetSizesAsMklDnnDims() in Compute()
218 ? grad_mkl_shape.GetSizesAsMklDnnDims() in Compute()
Dmkl_input_conversion_op.cc127 input_shape_0.GetSizesAsMklDnnDims(), in Compute()
228 mkl_shape->GetSizesAsMklDnnDims(), in Compute()
Dmkl_maxpooling_op.cc121 ? dnn_shape_input.GetSizesAsMklDnnDims() in Compute()
266 ? orig_input_mkl_shape.GetSizesAsMklDnnDims() in Compute()
274 ? grad_mkl_shape.GetSizesAsMklDnnDims() in Compute()
Dmkl_relu_op.cc468 src_dims = dnn_shape_src.GetSizesAsMklDnnDims(); in Compute()
505 dnn_shape_src.GetSizesAsMklDnnDims(), in Compute()
633 src_dims = dnn_shape_src.GetSizesAsMklDnnDims(); in Compute()
663 src_dims = dnn_shape_src.GetSizesAsMklDnnDims(); in Compute()
716 dnn_shape_src.GetSizesAsMklDnnDims(), in Compute()
721 dnn_shape_diff_dst.GetSizesAsMklDnnDims(), in Compute()
Dmkl_quantize_op.cc337 ? src_mkl_shape.GetSizesAsMklDnnDims() in Compute()
412 src_mkl_shape.GetSizesAsMklDnnDims(), in Compute()
Dmkl_lrn_op.cc131 memory::dims input_dims = src_dnn_shape.GetSizesAsMklDnnDims(); in Compute()
383 orig_input_dnn_shape.GetSizesAsMklDnnDims(); in Compute()
484 memory::dims orig_input_dims = input_grad_dnn_shape.GetSizesAsMklDnnDims(); in ConfigureInputGradient()
Dmkl_aggregate_ops.cc226 mkl_shape.GetSizesAsMklDnnDims(), in Compute()
Dmkl_softmax_op.cc218 src_dims = src_mkl_shape.GetSizesAsMklDnnDims(); in Compute()
Dmkl_fused_batch_norm_op.cc769 ? dnn_shape_src.GetSizesAsMklDnnDims() in Compute()
1163 ? dnn_shape_src.GetSizesAsMklDnnDims() in Compute()
1167 ? dnn_shape_diff_dst.GetSizesAsMklDnnDims() in Compute()
Dmkl_slice_op.cc419 input_dims = input_mkl_shape.GetSizesAsMklDnnDims(); in ComputeMklSlice()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h376 inline memory::dims GetSizesAsMklDnnDims() const { in GetSizesAsMklDnnDims() function