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 | 5100 const RuntimeShape ext_output_shape = in PadImpl() local
5118 const int output_batch = ext_output_shape.Dims(0); in PadImpl()
5119 const int output_height = ext_output_shape.Dims(1); in PadImpl()
5120 const int output_width = ext_output_shape.Dims(2); in PadImpl()
5121 const int output_depth = ext_output_shape.Dims(3); in PadImpl()
5144 TypedMemset<T>(output_data + Offset(ext_output_shape, out_b, 0, 0, 0), in PadImpl()
5151 output_data + Offset(ext_output_shape, out_b, out_h, 0, 0), in PadImpl()
5158 output_data + Offset(ext_output_shape, out_b, out_h, out_w, 0), in PadImpl()
5163 Offset(ext_output_shape, out_b, out_h, out_w, left_d_padding); in PadImpl()
5171 output_data + Offset(ext_output_shape, out_b, out_h, out_w, in PadImpl()
[all …]
|