Searched refs:crops_dims (Results 1 – 3 of 3) 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/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 1787 const int32* crops_data, const Dims<4>& crops_dims, in BatchToSpaceND() argument
1791 DimsToShape(crops_dims), crops_data, DimsToShape(output_dims), in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 1955 const int32* crops_data, const Dims<4>& crops_dims, in BatchToSpaceND() argument
1959 DimsToShape(crops_dims), crops_data, DimsToShape(output_dims), in BatchToSpaceND()
|