Searched refs:crops_dims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_batch_to_space_nd_attributes.cc | 51 const std::vector<int>& crops_dims = crops_array.shape().dims(); in Run() local
52 if (crops_dims.size() != 2) { in Run()
59 for (int i = 0; i < crops_dims[0]; ++i) { in Run()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | batch_to_space_nd_test.cc | 73 TfLiteIntArray* crops_dims = IntArrayFromInts(crops_dims_data); in TestBatchToSpaceNdFloat() local
82 CreateTensor(crops_data, crops_dims), in TestBatchToSpaceNdFloat()
100 TfLiteIntArray* crops_dims = IntArrayFromInts(crops_dims_data); in TestBatchToSpaceNdQuantized() local
111 tflite::testing::CreateTensor(crops_data, crops_dims), in TestBatchToSpaceNdQuantized()
|
| D | space_to_batch_nd_test.cc | 73 TfLiteIntArray* crops_dims = IntArrayFromInts(crops_dims_data); in TestSpaceToBatchNdFloat() local
82 CreateTensor(crops_data, crops_dims), in TestSpaceToBatchNdFloat()
100 TfLiteIntArray* crops_dims = IntArrayFromInts(crops_dims_data); in TestSpaceToBatchNdQuantized() local
111 tflite::testing::CreateTensor(crops_data, crops_dims), in TestSpaceToBatchNdQuantized()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 1079 int crops_dims = crops_type.getShape()[0]; in convertBatchToSpaceNDOp() local
1097 SmallVector<std::pair<int64_t, int64_t>, 4> crops(crops_dims); in convertBatchToSpaceNDOp()
1113 SmallVector<int32_t, 2> crops_const(2 * (crops_dims)); in convertBatchToSpaceNDOp()
1119 for (int i = 0; i < crops_dims; i++) { in convertBatchToSpaceNDOp()
1120 int crops_lo = crops_const[i * crops_dims + 0]; in convertBatchToSpaceNDOp()
1121 int crops_hi = crops_const[i * crops_dims + 1]; in convertBatchToSpaceNDOp()
1216 if (i == 0 || i > crops_dims) { in convertBatchToSpaceNDOp()
1221 assert(i - 1 >= 0 && i - 1 < crops_dims); in convertBatchToSpaceNDOp()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 2047 const int32* crops_data, const Dims<4>& crops_dims, in BatchToSpaceND() argument
2051 DimsToShape(crops_dims), crops_data, DimsToShape(output_dims), in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4845 const int32* crops_data, const Dims<4>& crops_dims, in BatchToSpaceND() argument
4849 DimsToShape(crops_dims), crops_data, DimsToShape(output_dims), in BatchToSpaceND()
|