Searched defs: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()
166 …ype> compute_target(TensorShape shape, unsigned int axis, std::vector<TensorShape> split_shapes, D… in compute_target()
218 …> compute_reference(TensorShape shape, unsigned int axis, std::vector<TensorShape> split_shapes, D… in compute_reference()
|
| /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()
|
| /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
|