Home
last modified time | relevance | path

Searched refs:multiples_array (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtile_ops.cc183 const gtl::ArraySlice<Tmultiples> multiples_array( in Compute() local
188 context, multiples_array[i] >= 0, in Compute()
190 multiples_array[i])); in Compute()
192 input.dim_size(i) * multiples_array[i])); in Compute()
206 HandleCase<DT>(context, multiples_array, result); \ in Compute()
244 const gtl::ArraySlice<Tmultiples> multiples_array, in HandleCaseImpl() argument
249 multiples_array); in HandleCaseImpl()
254 const gtl::ArraySlice<Tmultiples> multiples_array,
263 OpKernelContext* context, const gtl::ArraySlice<Tmultiples> multiples_array, in HandleCase() argument
277 const gtl::ArraySlice<Tmultiples> multiples_array, Tensor* result) { \
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dresolve_constant_tile.cc73 inline void Tile(const Array& input_array, const Array& multiples_array, in Tile() argument
79 switch (multiples_array.data_type) { in Tile()
83 multiples_array.GetBuffer<ArrayDataType::kInt32>().data.data(), in Tile()
89 multiples_array.GetBuffer<ArrayDataType::kInt64>().data.data(), in Tile()
130 const Array& multiples_array = model->GetArray(op->inputs[1]); in Run() local
131 CHECK(multiples_array.data_type == ArrayDataType::kInt32 || in Run()
132 multiples_array.data_type == ArrayDataType::kInt64) in Run()
140 Tile<ArrayDataType::kFloat>(input_array, multiples_array, &output_array); in Run()
143 Tile<ArrayDataType::kUint8>(input_array, multiples_array, &output_array); in Run()
146 Tile<ArrayDataType::kInt16>(input_array, multiples_array, &output_array); in Run()
[all …]
H A Dconvert_trivial_tile_to_concat.cc35 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()
H A Dpropagate_fixed_sizes.cc1922 auto& multiples_array = model->GetArray(op->inputs[1]); in ProcessTileOperator() local
1923 if (!multiples_array.has_shape()) { in ProcessTileOperator()
1927 if (!multiples_array.buffer) { in ProcessTileOperator()
1931 CHECK(multiples_array.data_type == ArrayDataType::kInt32) in ProcessTileOperator()
1935 multiples_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessTileOperator()