| /external/ComputeLibrary/src/graph/nodes/ |
| 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()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| 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 …]
|
| D | sharding_util_ops.cc | 368 const std::vector<int32>& num_splits, in ValidateShapesForSlice()
428 SliceAndMaybePadState(absl::Span<const int32> num_splits, in SliceAndMaybePadState()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_conversion_ops.cc | 161 int64_t num_splits; in RaggedTensorToSparseShapeFn() local
190 int64_t num_splits; in RaggedTensorToVariantShapeFn() local
|
| D | ragged_array_ops.cc | 136 int num_splits; in RaggedGatherShapeFn() local
|
| D | parsing_ops.cc | 68 DimensionHandle num_splits; in AddRaggedOutputShapes() local
|
| D | sparse_ops.cc | 337 int num_splits = c->num_outputs() / 3; in __anonc55240e60e02() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| 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()
|
| /external/tensorflow/tensorflow/core/tpu/ops/ |
| D | tpu_partitioned_output_op.cc | 37 int num_splits; in __anon141003b80102() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| 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()
|
| D | split_v_test.cc | 42 int num_splits, int axis, in SplitVOpModel()
97 int num_splits = size_splits_data.size(); in Check() local
|
| D | split.cc | 53 const TfLiteTensor* input, int num_splits) { in ResizeOutputTensors()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | split_tester.h | 51 inline SplitTester& NumSplits(int num_splits) { in NumSplits()
|
| /external/ComputeLibrary/src/graph/mutators/ |
| D | SplitLayerSubTensorMutator.cpp | 86 const unsigned int num_splits = split_node->num_splits(); in mutate() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_variant_op_test.h | 50 int64_t num_splits = ragged_splits.size(); in BuildEncodeRaggedTensorGraph() local
|
| 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
|
| D | ragged_tensor_to_sparse_kernel_test.cc | 44 int64_t num_splits = rt_nested_splits.size(); in BuildRaggedTensorToSparseGraph() local
|
| D | ragged_gather_op_test.cc | 42 int64_t num_splits = PARAMS_RAGGED_RANK + indices_shape.dims() - 1; in BuildRaggedGatherGraph() local
|
| D | ragged_tensor_to_variant_op.cc | 82 int num_splits = ragged_rank - 1; in UnbatchRaggedZerothDim() local
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| D | sentencepiece_tokenizer_op.cc | 56 tensorflow::shape_inference::DimensionHandle num_splits; in __anon9b54717e0102() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | multi_output_op.json | 79 "num_splits": 2 number
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| 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
|
| /external/ComputeLibrary/examples/ |
| D | graph_deepspeech_v0_4_1.cpp | 270 const unsigned int num_splits = 4; in add_lstm_cell() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | device_transform_patterns.cc | 155 int num_splits, PatternRewriter* rewriter) { in SliceOutputs()
289 auto num_splits = split_op.num_splits(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 2303 const int64_t num_splits = ctrl_.number_of_splits; in SelectValidPortion() local
2596 const int64_t num_splits = ctrl_.number_of_splits; in PropagateOnConv() local
2858 int64_t num_splits) { in PerformSplitSpace()
2904 int64_t spatial_split_size, int64_t num_splits) { in PadAndSplitSpace()
2938 int64_t spatial_split_size, int64_t num_splits, in SplitSpace()
|