Searched refs:split_shapes (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | SplitFixture.h | 153 …void setup(TensorShape shape, unsigned int axis, std::vector<TensorShape> split_shapes, DataType d… in setup() argument
155 _target = compute_target(shape, axis, split_shapes, data_type); in setup()
156 _reference = compute_reference(shape, axis, split_shapes, data_type); in setup()
166 …target(TensorShape shape, unsigned int axis, std::vector<TensorShape> split_shapes, DataType data_… in compute_target() argument
173 for(const auto &split_shape : split_shapes) in compute_target()
218 …erence(TensorShape shape, unsigned int axis, std::vector<TensorShape> split_shapes, DataType data_… in compute_reference() argument
228 for(const auto &split_shape : split_shapes) in compute_reference()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | conv_2d_helpers.cc | 192 std::vector<RuntimeShape> split_shapes; in SplitWeightsForDwConv() local
196 split_shapes.reserve(per_channel_quant_.splits); in SplitWeightsForDwConv()
201 split_shapes.push_back({weights_height_size, weights_width_size, in SplitWeightsForDwConv()
204 split_shapes_data.push_back(&split_shapes.back()); in SplitWeightsForDwConv()
217 split_shapes[s].DimsData(), in SplitWeightsForDwConv()
|
| /aosp_15_r20/external/ComputeLibrary/tests/datasets/ |
| H A D | SplitDataset.h | 186 void add_config(TensorShape shape, unsigned int axis, std::vector<TensorShape> split_shapes) in add_config() argument
190 _split_shapes_values.emplace_back(split_shapes); in add_config()
|