| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | sharding_util_ops_test.cc | 78 const std::vector<int32> num_splits = {2, 2}; in TEST() local
116 const std::vector<int32> num_splits = {2, 2}; in TEST() local
133 absl::Span<const int32> num_splits, in CreateSplitTensorGraph()
160 absl::Span<const int32> num_splits, in CreateSplitResourceGraph()
215 const std::vector<int32> num_splits = {1, 1, 0}; in TEST_P() local
233 const std::vector<int32> num_splits = {1, -1, 1}; in TEST_P() local
251 const std::vector<int32> num_splits = {2}; in TEST_P() local
269 const std::vector<int32> num_splits = {2, 2}; in TEST_P() local
286 const std::vector<int32> num_splits = {2, 2}; in TEST_P() local
304 const std::vector<int32> num_splits = {2}; in TEST_P() local
[all …]
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | split_quant8_signed.mod.py | 19 num_splits = Int32Scalar("num_splits", 3) variable
42 num_splits = Int32Scalar("num_splits", 2) variable
63 num_splits = Int32Scalar("num_splits", 3) variable
86 num_splits = Int32Scalar("num_splits", 2) variable
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/nodes/ |
| H A D | SplitLayerNode.cpp | 35 SplitLayerNode::SplitLayerNode(unsigned int num_splits, int axis, std::vector<int> size_splits) in SplitLayerNode()
42 unsigned int SplitLayerNode::num_splits() const in num_splits() function in arm_compute::graph::SplitLayerNode
53 … unsigned int num_splits, int axis, unsigned int idx) in compute_output_descriptor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | ragged_conversion_ops.cc | 161 int64_t num_splits; in RaggedTensorToSparseShapeFn() local
190 int64_t num_splits; in RaggedTensorToVariantShapeFn() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| H A D | split_test.cc | 27 int num_splits, int axis) { in SplitOpModel()
59 int axis, int num_splits, std::initializer_list<int> input_shape, in CheckSplitBehavior()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | split_test.cc | 41 SplitOpModel(const TensorData& input, int num_splits, in SplitOpModel()
80 void Check(TestType test_type, int axis, int num_splits, in Check()
|
| H A D | split_v_test.cc | 42 int num_splits, int axis, in SplitVOpModel()
97 int num_splits = size_splits_data.size(); in Check() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
| H A D | flash_api.cpp | 274 auto is_split_eligible = [&ceildiv, &num_n_blocks](int num_splits) { in num_splits_heuristic()
277 for (int num_splits = 1; num_splits <= max_splits; num_splits++) { in num_splits_heuristic() local
288 for (int num_splits = 1; num_splits <= max_splits; num_splits++) { in num_splits_heuristic() local
300 const int num_splits, cudaDeviceProp *dprops, struct c10::TensorOptions opts) { in set_params_splitkv()
1270 int num_splits in mha_fwd_kvcache()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | ragged_gather_op.cc | 130 int num_splits = indices_in.dims() - 1 + params_nested_splits_in.size(); in MakeSplits() local
230 SPLITS_TYPE num_splits = out_splits[i].size(); in WriteSplits() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | xla_sharding_util.cc | 153 const int num_splits = num_splits_and_index.value(); in HandleTileShardedInputs() local
441 for (auto num_splits : llvm::reverse(sharding.tile_assignment_dimensions())) { in HandleTileShardedOutputs() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | split_int32_2.mod.py | 20 num_splits = Int32Scalar("num_splits", 2) variable
|
| D | split_quant8_2.mod.py | 20 num_splits = Int32Scalar("num_splits", 2) variable
|
| D | split_float_5.mod.py | 20 num_splits = Int32Scalar("num_splits", 2) variable
|
| D | split_int32_4.mod.py | 20 num_splits = Int32Scalar("num_splits", 2) variable
|
| D | split_quant8_4.mod.py | 20 num_splits = Int32Scalar("num_splits", 2) variable
|
| D | split_float_2.mod.py | 20 num_splits = Int32Scalar("num_splits", 2) variable
|
| D | split_float_4.mod.py | 20 num_splits = Int32Scalar("num_splits", 2) variable
|
| D | split_float_1.mod.py | 20 num_splits = Int32Scalar("num_splits", 3) variable
|
| D | split_int32_1.mod.py | 20 num_splits = Int32Scalar("num_splits", 3) variable
|
| D | split_quant8_1.mod.py | 20 num_splits = Int32Scalar("num_splits", 3) variable
|
| D | split_quant8_3.mod.py | 20 num_splits = Int32Scalar("num_splits", 3) variable
|
| D | split_float_3.mod.py | 20 num_splits = Int32Scalar("num_splits", 3) variable
|
| D | split_int32_3.mod.py | 20 num_splits = Int32Scalar("num_splits", 3) variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ops/ |
| H A D | tpu_partitioned_output_op.cc | 37 int num_splits; in __anon0352b5690102() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | split_tester.h | 51 inline SplitTester& NumSplits(int num_splits) { in NumSplits()
|