Searched refs:multiples_array (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_tile.cc | 72 inline void Tile(const Array& input_array, const Array& multiples_array, in Tile() argument
78 switch (multiples_array.data_type) { in Tile()
82 multiples_array.GetBuffer<ArrayDataType::kInt32>().data.data(), in Tile()
88 multiples_array.GetBuffer<ArrayDataType::kInt64>().data.data(), in Tile()
129 const Array& multiples_array = model->GetArray(op->inputs[1]); in Run() local
130 CHECK(multiples_array.data_type == ArrayDataType::kInt32 || in Run()
131 multiples_array.data_type == ArrayDataType::kInt64) in Run()
139 Tile<ArrayDataType::kFloat>(input_array, multiples_array, &output_array); in Run()
142 Tile<ArrayDataType::kUint8>(input_array, multiples_array, &output_array); in Run()
145 Tile<ArrayDataType::kInt16>(input_array, multiples_array, &output_array); in Run()
[all …]
|
| D | convert_trivial_tile_to_concat.cc | 35 const auto& multiples_array = model->GetArray(tile_op->inputs[1]); in Run() local
37 if (!input_array.has_shape() || !multiples_array.has_shape() || in Run()
44 if (!multiples_array.buffer) { in Run()
49 multiples_array.GetBuffer<ArrayDataType::kInt32>().data; in Run()
|
| D | propagate_fixed_sizes.cc | 1898 auto& multiples_array = model->GetArray(op->inputs[1]); in ProcessTileOperator() local
1899 if (!multiples_array.has_shape()) { in ProcessTileOperator()
1903 if (!multiples_array.buffer) { in ProcessTileOperator()
1907 CHECK(multiples_array.data_type == ArrayDataType::kInt32) in ProcessTileOperator()
1911 multiples_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessTileOperator()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tile_ops.cc | 109 const gtl::ArraySlice<Tmultiples> multiples_array( in Compute() local
114 context, multiples_array[i] >= 0, in Compute()
116 multiples_array[i])); in Compute()
117 output_shape.AddDim(input.dim_size(i) * multiples_array[i]); in Compute()
131 HandleCase<DT>(context, multiples_array, result); \ in Compute()
165 const gtl::ArraySlice<Tmultiples>& multiples_array, in HandleCaseImpl() argument
170 multiples_array); in HandleCaseImpl()
175 const gtl::ArraySlice<Tmultiples>& multiples_array,
185 const gtl::ArraySlice<Tmultiples>& multiples_array, Tensor* result) { in HandleCase() argument
198 const gtl::ArraySlice<Tmultiples>& multiples_array, Tensor* result) { \
[all …]
|