Searched refs:ext_output_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5879 const RuntimeShape ext_output_shape = in PadImpl() local
5897 const int output_batch = ext_output_shape.Dims(0); in PadImpl()
5898 const int output_height = ext_output_shape.Dims(1); in PadImpl()
5899 const int output_width = ext_output_shape.Dims(2); in PadImpl()
5900 const int output_depth = ext_output_shape.Dims(3); in PadImpl()
5923 TypedMemset<T>(output_data + Offset(ext_output_shape, out_b, 0, 0, 0), in PadImpl()
5930 output_data + Offset(ext_output_shape, out_b, out_h, 0, 0), in PadImpl()
5937 output_data + Offset(ext_output_shape, out_b, out_h, out_w, 0), in PadImpl()
5942 Offset(ext_output_shape, out_b, out_h, out_w, left_d_padding); in PadImpl()
5950 output_data + Offset(ext_output_shape, out_b, out_h, out_w, in PadImpl()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 6204 const RuntimeShape ext_output_shape = in PadImpl() local
6222 const int output_batch = ext_output_shape.Dims(0); in PadImpl()
6223 const int output_height = ext_output_shape.Dims(1); in PadImpl()
6224 const int output_width = ext_output_shape.Dims(2); in PadImpl()
6225 const int output_depth = ext_output_shape.Dims(3); in PadImpl()
6248 TypedMemset<T>(output_data + Offset(ext_output_shape, out_b, 0, 0, 0), in PadImpl()
6255 output_data + Offset(ext_output_shape, out_b, out_h, 0, 0), in PadImpl()
6262 output_data + Offset(ext_output_shape, out_b, out_h, out_w, 0), in PadImpl()
6267 Offset(ext_output_shape, out_b, out_h, out_w, left_d_padding); in PadImpl()
6275 output_data + Offset(ext_output_shape, out_b, out_h, out_w, in PadImpl()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 3200 const RuntimeShape ext_output_shape = in PadImpl() local
3216 const int output_batch = ext_output_shape.Dims(0); in PadImpl()
3217 const int output_height = ext_output_shape.Dims(1); in PadImpl()
3218 const int output_width = ext_output_shape.Dims(2); in PadImpl()
3219 const int output_depth = ext_output_shape.Dims(3); in PadImpl()
|