| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_conversion_ops.cc | 160 int64 num_splits; in RaggedTensorToSparseShapeFn() local
189 int64 num_splits; in RaggedTensorToVariantShapeFn() local
|
| D | ragged_array_ops.cc | 122 int num_splits; in RaggedGatherShapeFn() local
|
| D | parsing_ops.cc | 68 DimensionHandle num_splits; in AddRaggedOutputShapes() local
|
| D | sparse_ops.cc | 333 int num_splits = c->num_outputs() / 3; in __anon36a8fc880e02() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | split_test.cc | 24 int num_splits, int axis) { in SplitOpModel()
56 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 __anon4952381a0102() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | split_v_test.cc | 36 int num_splits, int axis) { in SplitVOpModel()
86 int num_splits = size_splits_data.size(); in Check() local
|
| 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.cc | 53 const TfLiteTensor* input, int num_splits) { in ResizeOutputTensors()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_gather_op.cc | 124 int num_splits = indices_in.dims() - 1 + params_nested_splits_in.size(); in MakeSplits() local
224 SPLITS_TYPE num_splits = out_splits[i].size(); in WriteSplits() local
|
| D | ragged_tensor_to_sparse_kernel_test.cc | 44 int64 num_splits = rt_nested_splits.size(); in BuildRaggedTensorToSparseGraph() local
|
| D | ragged_gather_op_test.cc | 42 int64 num_splits = PARAMS_RAGGED_RANK + indices_shape.dims() - 1; in BuildRaggedGatherGraph() local
|
| D | ragged_tensor_to_variant_op.cc | 42 int num_splits = ragged_rank - 1; in UnbatchRaggedZerothDim() local
|
| D | ragged_tensor_to_variant_op_test.cc | 47 int64 num_splits = ragged_splits.size(); in BuildEncodeRaggedTensorGraph() local
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| D | sentencepiece_tokenizer_op.cc | 56 tensorflow::shape_inference::DimensionHandle num_splits; in __anon18bebb200102() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 159 const int num_splits = num_splits_and_index.value(); in HandleTileShardedInputs() local
427 for (auto num_splits : llvm::reverse(sharding.tile_assignment_dimensions())) { in HandleTileShardedOutputs() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 1665 const int64 num_splits = new_batch_size / old_batch_size; in SelectValidPortion() local
1908 const int64 num_splits = kNumSplits; in PropagateOnConv() local
2040 int64 spatial_split_size, int64 num_splits) { in SplitSpaceHelper()
2096 int64 num_splits, bool is_backprop, in SplitSpace()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 712 xla::StatusOr<Node*> CreateSplitNode(const int num_splits, const int dim, in CreateSplitNode()
777 int64 GetPadding(const int split_dim, const int num_splits, in GetPadding()
883 int num_splits; in CreateOrGetSplitNodesForInputSharding() local
906 xla::StatusOr<Node*> CreateConcatNode(int dim, int num_splits, DataType dtype, in CreateConcatNode()
1033 auto num_splits = it->second; in CreateConcatNodesForRetval() local
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 379 int num_splits; member
383 int num_splits; member
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1855 Operation *op, int64_t num_splits, in VerifySplitOpOutputTypes()
1869 int64_t num_splits = op.num_splits(); in Verify() local
1905 int64_t num_splits = op.num_splits(); in Verify() local
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_split_vars.c | 367 unsigned num_splits; member
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4963 int num_splits, bool squeeze_after) { in ConvertSplitHelper()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2931 int64_t num_splits = op.getNumResults(); in matchAndRewrite() local
|