| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_v_op_test.cc | 60 static Graph* MakeGraph(int split_dim, const std::vector<int64_t>& size_splits, in MakeGraph()
93 #define BM_SPLITV_2D(split_dim, num_split, dim0, dim1) \ argument
110 #define BM_SPLITV_3D(split_dim, num_split, dim0, dim1, dim2) \ argument
|
| D | split_op_test.cc | 29 static Graph* MakeGraph(int split_dim, int num_split, in MakeGraph()
59 #define BM_SPLIT_2D(split_dim, num_split, chunk_size0, chunk_size1) \ argument
|
| D | split_op.cc | 56 const int32_t split_dim = in ComputeEasyCases() local
133 const TensorShape& input_shape, int32_t split_dim, in operator ()()
214 const int32_t split_dim = in Compute() local
284 const int32_t split_dim = in Compute() local
|
| D | split_v_op.cc | 66 const int32_t split_dim = in ComputeEasyCases() local
198 const TensorShape& input_shape, int32_t split_dim, in SplitHasAlignedOutputsInFirstDimension()
221 const TensorShape& input_shape, int32_t split_dim, in operator ()()
307 const int32_t split_dim = in Compute() local
386 const int32_t split_dim = in Compute() local
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | split_tester.h | 36 inline SplitTester& SplitDimension(int32_t split_dim) { in SplitDimension()
62 int32_t split_dim = SplitDimension(); in OutputShape() local
|
| D | split_tester.cc | 163 std::array<int32_t, 1> split_dim = {SplitDimension()}; in CreateTfLiteModel() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | split_op.cc | 49 int32_t split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims() in Compile() local
115 int64_t split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims() in Compile() local
|
| /external/XNNPACK/src/subgraph/ |
| D | even-split.c | 397 size_t split_dim, in check_output_value()
478 size_t split_dim, in xnn_define_even_split_n()
625 size_t split_dim, in xnn_define_even_split2()
638 size_t split_dim, in xnn_define_even_split3()
652 size_t split_dim, in xnn_define_even_split4()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reduce_scatter_utils.h | 26 int64_t split_dim; member
|
| D | all_to_all_decomposer.cc | 58 int64_t split_dim = *all_to_all->split_dimension(); in ExpandInstruction() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_reduce_scatter_creator.cc | 58 const int64_t split_dim = ar_spec->split_dim; in Run() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | device_transform_patterns.cc | 154 RankedTensorType input_type, int split_dim, in SliceOutputs()
304 int split_dim = -1; in matchAndRewrite() local
325 auto split_dim = splitv_op.split_dim().getDefiningOp(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | split_spmd_expander.cc | 40 int64_t split_dim, const llvm::DenseMap<int, Layout>& layouts) { in MergeLayoutsForSplitOutput()
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_util_test.py | 350 def test_blockwise_input(self, op_dimension_values, split_dim): argument
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradient_checker_test.cc | 145 auto split_dim = Const(scope, 1, {}); in TEST() local
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 497 const int split_dim, const int num_split, in Split()
|
| /external/ComputeLibrary/src/cpu/operators/internal/ |
| D | CpuGemmAssemblyDispatch.cpp | 537 const unsigned int split_dim = scheduling_hint.split_dimension(); in run() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 984 split_dim=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 163 int64_t split_dim; member
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 588 int64_t split_dim = c->Value(split_dimension); in __anon091a71b70a02() local
649 int64_t split_dim = c->Value(split_dimension); in __anon091a71b70b02() local
|
| D | array_ops_test.cc | 1204 Tensor split_dim = test::AsTensor<int32>({1, 2}); in TEST() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_functional_ops.cc | 1456 std::vector<NodeDef>& ndefs, int split_dim, in InitializeShardedVarOnTPU()
1821 int split_dim = -1; in ReplaceAndPartitionXLAShardingVariable() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 1025 uint64_t split_dim = all_to_all->split_dimension().value(); in ImportInstructionImpl() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 105 auto split_dim = Const(scope_, 1, {}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2373 int64_t split_dim = split_dim_opt.getValue(); in verify() local
2410 int64_t split_dim = split_dim_opt.getValue(); in verify() local
|