Searched refs:ext_output_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | pad.h | 44 const RuntimeShape ext_output_shape = in PadImpl() local
69 const int output_batch = ext_output_shape.Dims(0); in PadImpl()
70 const int output_height = ext_output_shape.Dims(1); in PadImpl()
71 const int output_width = ext_output_shape.Dims(2); in PadImpl()
72 const int output_depth = ext_output_shape.Dims(3); in PadImpl()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5024 const RuntimeShape ext_output_shape = in PadImpl() local
5042 const int output_batch = ext_output_shape.Dims(0); in PadImpl()
5043 const int output_height = ext_output_shape.Dims(1); in PadImpl()
5044 const int output_width = ext_output_shape.Dims(2); in PadImpl()
5045 const int output_depth = ext_output_shape.Dims(3); in PadImpl()
5068 TypedMemset<T>(output_data + Offset(ext_output_shape, out_b, 0, 0, 0), in PadImpl()
5075 output_data + Offset(ext_output_shape, out_b, out_h, 0, 0), in PadImpl()
5082 output_data + Offset(ext_output_shape, out_b, out_h, out_w, 0), in PadImpl()
5087 Offset(ext_output_shape, out_b, out_h, out_w, left_d_padding); in PadImpl()
5095 output_data + Offset(ext_output_shape, out_b, out_h, out_w, in PadImpl()
[all …]
|