Searched refs:in_dimensions (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | tile.cc | 107 std::pair<int, int> TileOneDimension(const TfLiteIntArray& in_dimensions, in TileOneDimension() argument
110 const int dimension_size = in_dimensions.data[dimension]; in TileOneDimension()
111 if (dimension == in_dimensions.size - 1) { in TileOneDimension()
124 TileOneDimension(in_dimensions, copy_from_data, multipliers, in TileOneDimension()
141 const TfLiteIntArray& in_dimensions, const TfLiteTensor* in_data, in TileStringOneDimension() argument
144 const int dimension_size = in_dimensions.data[dimension]; in TileStringOneDimension()
145 if (dimension == in_dimensions.size - 1) { in TileStringOneDimension()
156 in_dimensions, in_data, in_data_index + total_stride_size, multipliers, in TileStringOneDimension()
172 void Tile(const TfLiteIntArray& in_dimensions, const TfLiteTensor* in_data, in Tile() argument
177 TileOneDimension(in_dimensions, GetTensorData<T>(in_data), in Tile()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_tile.cc | 40 std::pair<int, int> TileOneDimension(const Shape& in_dimensions, in TileOneDimension() argument
43 const int dimension_size = in_dimensions.dims(dimension); in TileOneDimension()
44 if (dimension == in_dimensions.dimensions_count() - 1) { in TileOneDimension()
57 TileOneDimension(in_dimensions, copy_from_data, multipliers, in TileOneDimension()
|
| /external/llvm-project/polly/lib/Transform/ |
| D | MaximalStaticExpansion.cpp | 353 unsigned in_dimensions = CurrentAccessMap.dim(isl::dim::in); in expandAccess() local
362 NewAccessMap = NewAccessMap.add_dims(isl::dim::out, in_dimensions); in expandAccess()
378 for (unsigned i = 0; i < in_dimensions; i++) { in expandAccess()
|