Home
last modified time | relevance | path

Searched refs:GetInputSizeInMklOrder (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h114 virtual inline void GetInputSizeInMklOrder(const TensorShape& input_shape, in GetInputSizeInMklOrder() function
540 GetInputSizeInMklOrder(input_shape, input_dims);
Dmkl_conv_grad_input_ops.cc408 conv_util.GetInputSizeInMklOrder(diff_dst_tf_shape, &diff_dst_dims); in Compute()
Dmkl_conv_grad_filter_ops.cc445 conv_util.GetInputSizeInMklOrder(diff_dst_tf_shape, &diff_dst_dims); in Compute()