Home
last modified time | relevance | path

Searched defs:src_mkl_shape (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_aggregate_ops.cc44 MklDnnShape src_mkl_shape; in GetTensorShape() local
75 MklDnnShape src_mkl_shape; in FindMKLInputIndex() local
184 MklDnnShape src_mkl_shape; in Compute() local
Dmkl_dequantize_op.cc63 MklDnnShape src_mkl_shape; in Compute() local
Dmkl_matmul_op_fused.cc63 MklDnnShape src_mkl_shape; in Compute() local
Dmkl_softmax_op.cc206 MklDnnShape src_mkl_shape; in Compute() local
Dmkl_qmatmul_op.cc167 MklDnnShape src_mkl_shape, weight_mkl_shape; in Compute() local
Dmkl_quantize_op.cc332 MklDnnShape src_mkl_shape; in Compute() local
Dmkl_conv_grad_input_ops.cc319 MklDnnShape src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape; in Compute() local
Dmkl_conv_grad_filter_ops.cc376 MklDnnShape src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape; in Compute() local
Dmkl_conv_ops.cc502 MklDnnShape src_mkl_shape, filter_mkl_shape; in Compute() local
/external/tensorflow/tensorflow/core/util/
Dmkl_types.h39 #define GET_FORMAT_FROM_SHAPE(src_mkl_shape) MklTensorFormat::FORMAT_BLOCKED argument