Searched refs:ext_input_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5877 const RuntimeShape ext_input_shape = in PadImpl() local
5912 const int input_depth = ext_input_shape.Dims(3); in PadImpl()
5944 Offset(ext_input_shape, out_b - left_b_padding, in PadImpl()
6027 const RuntimeShape ext_input_shape = in PadImageStyleMemset() local
6053 const int batch = MatchingDim(ext_input_shape, 0, ext_output_shape, 0); in PadImageStyleMemset()
6056 const int input_height = ext_input_shape.Dims(1); in PadImageStyleMemset()
6057 const int input_width = ext_input_shape.Dims(2); in PadImageStyleMemset()
6058 const int depth = MatchingDim(ext_input_shape, 3, ext_output_shape, 3); in PadImageStyleMemset()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 6202 const RuntimeShape ext_input_shape = in PadImpl() local
6237 const int input_depth = ext_input_shape.Dims(3); in PadImpl()
6269 Offset(ext_input_shape, out_b - left_b_padding, in PadImpl()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 3198 const RuntimeShape ext_input_shape = in PadImpl() local
|