Cross Reference: /external/tensorflow/tensorflow/core/kernels/mkl_conv

Lines Matching defs:in_dims

208     mkl_context.in_dims = input_in_mkl_format
260               mkl_context.in_dims, mkl_context.in_sizes, mkl_context.out_sizes,
268               mkl_context.in_dims, mkl_context.in_sizes, mkl_context.out_sizes,
278     mkl_output_mkl_shape.SetTfLayout(mkl_context.in_dims, mkl_context.out_sizes,
282     mkl_output_mkl_shape.SetTfDimOrder(mkl_context.in_dims, data_format_);
333     int in_dims;
357         CHECK_EQ(dnnLayoutCreate_F32(&lt_input, in_dims, in_sizes, in_strides),

OpenGrok