Searched defs:input_rows_raw (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_conv_ops.h | 134 int64 input_rows_raw = GetTensorDim(input_shape, data_format_, 'H'); in GetInputSizeInMklOrder() local
157 int64 input_rows_raw = GetTensorDim(input_shape, data_format_, '1'); in GetInputSizeInMklOrder() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | depthwise_conv_op.cc | 376 const int64_t input_rows_raw = GetTensorDim(input, data_format_, 'H'); in Compute() local
|
| H A D | conv_ops_using_gemm.cc | 492 const int64_t input_rows_raw = GetTensorDim(input, data_format_, 'H'); in Compute() local
|
| H A D | conv_ops.cc | 586 const int64_t input_rows_raw = GetTensorDim(input, params.data_format, 'H'); in ComputeConv2DDimension() local
|