| /external/tensorflow/tensorflow/core/kernels/ |
| 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_v_op_test.cc | 57 static Graph* MakeGraph(int split_dim, const std::vector<int64>& size_splits, in MakeGraph()
90 #define BM_SPLITV_2D(split_dim, num_split, total_size0, total_size1) \ argument
|
| D | split_op.cc | 56 const int32 split_dim = in ComputeEasyCases() local
133 const TensorShape& input_shape, int32 split_dim, in operator ()()
214 const int32 split_dim = in Compute() local
284 const int32 split_dim = in Compute() local
|
| D | split_v_op.cc | 66 const int32 split_dim = in ComputeEasyCases() local
191 const TensorShape& input_shape, int32 split_dim, in operator ()()
272 const int32 split_dim = in Compute() local
351 const int32 split_dim = in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | split_op.cc | 48 int32 split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims() in Compile() local
114 int64 split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims() in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_to_all_decomposer.cc | 58 int64 split_dim = *all_to_all->split_dimension(); in ExpandInstruction() 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 | 785 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/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 991 split_dim=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 165 int64 split_dim; member
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 105 auto split_dim = Const(scope_, 1, {}); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 587 int64 split_dim = c->Value(split_dimension); in __anon42d741190a02() local
648 int64 split_dim = c->Value(split_dimension); in __anon42d741190b02() local
|
| D | array_ops_test.cc | 1193 Tensor split_dim = test::AsTensor<int32>({1, 2}); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1882 int64_t split_dim = split_dim_opt.getValue(); in Verify() local
1918 int64_t split_dim = split_dim_opt.getValue(); in Verify() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 663 const int split_dim, DataType dtype, in CreatePadNode()
777 int64 GetPadding(const int split_dim, const int num_splits, in GetPadding()
|
| /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/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 | 2066 auto split_dim = ops::Const(scope.WithOpName("split_dim"), {0}, {}); in TEST_F() local
2110 auto split_dim = ops::Const(scope.WithOpName("split_dim"), {0}, {}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1402 Value split_dim = op.split_dim(); in VerifySplitInputAndSplitDim() local
|