Searched refs:ext_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 3384 const RuntimeShape ext_shape = RuntimeShape::ExtendedShape(4, input_shape); in Slice() local
3393 ? ext_shape.Dims(0) - start_b in Slice()
3397 ? ext_shape.Dims(1) - start_h in Slice()
3401 ? ext_shape.Dims(2) - start_w in Slice()
3405 ? ext_shape.Dims(3) - start_d in Slice()
3413 *out_ptr++ = input_data[Offset(ext_shape, in_b, in_h, in_w, in_d)]; in Slice()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 6168 const RuntimeShape ext_shape = RuntimeShape::ExtendedShape(4, input_shape); in Slice() local
6177 ? ext_shape.Dims(0) - start_b in Slice()
6181 ? ext_shape.Dims(1) - start_h in Slice()
6185 ? ext_shape.Dims(2) - start_w in Slice()
6189 ? ext_shape.Dims(3) - start_d in Slice()
6198 input_data + Offset(ext_shape, in_b, in_h, in_w, start_d), in Slice()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 6337 const RuntimeShape ext_shape = RuntimeShape::ExtendedShape(4, input_shape); in Slice() local
6346 ? ext_shape.Dims(0) - start_b in Slice()
6350 ? ext_shape.Dims(1) - start_h in Slice()
6354 ? ext_shape.Dims(2) - start_w in Slice()
6358 ? ext_shape.Dims(3) - start_d in Slice()
6367 input_data + Offset(ext_shape, in_b, in_h, in_w, start_d), in Slice()
|