| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_v_op_test.cc | 59 static Graph* MakeGraph(int split_dim, const std::vector<int64>& size_splits, in MakeGraph()
92 #define BM_SPLITV_2D(split_dim, num_split, dim0, dim1) \ argument
109 #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
191 const TensorShape& input_shape, int32_t split_dim, in SplitHasAlignedOutputsInFirstDimension()
264 const TensorShape& input_shape, int32_t split_dim, in operator ()()
361 const int32_t split_dim = in Compute() local
440 const int32_t split_dim = in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | split_op.cc | 48 int32_t split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims() in Compile() local
114 int64_t split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims() in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reduce_scatter_utils.h | 24 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 | 55 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 | 150 RankedTensorType input_type, int split_dim, in SliceOutputs()
300 int split_dim = -1; in matchAndRewrite() local
321 auto split_dim = splitv_op.split_dim().getDefiningOp(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_util_test.py | 354 def test_blockwise_input(self, op_dimension_values, split_dim): argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local.py | 793 def make_2d(tensor, 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 | 496 const int split_dim, const int num_split, in Split()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEGEMMAssemblyDispatch.cpp | 639 const unsigned int split_dim = scheduling_hint.split_dimension(); in run() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 988 split_dim=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 165 int64 split_dim; member
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 587 int64_t split_dim = c->Value(split_dimension); in __anon5ce23fde0a02() local
648 int64_t split_dim = c->Value(split_dimension); in __anon5ce23fde0b02() 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 | 1403 std::vector<NodeDef>& ndefs, int split_dim, int device_ordinal) { in InitializeShardedVarOnTPU()
1766 int split_dim = -1; in ReplaceAndPartitionXLAShardingVariable() 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 | 2194 int64_t split_dim = split_dim_opt.getValue(); in Verify() local
2230 int64_t split_dim = split_dim_opt.getValue(); in Verify() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 790 const int split_dim = input_shape.dims(axis); in ProcessTensorFlowSplitOperator() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 676 const int split_dim, DataType dtype, in CreatePadNode()
790 int64 GetPadding(const int split_dim, const int num_splits, in GetPadding()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1165 const int split_dim = split_dim_data[0]; in ConvertSplitOperator() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 2071 auto split_dim = ops::Const(scope.WithOpName("split_dim"), {0}, {}); in TEST_F() local
2115 auto split_dim = ops::Const(scope.WithOpName("split_dim"), {0}, {}); in TEST_F() local
|