| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_op_test.cc | 29 static Graph* MakeGraph(int split_dim, int num_split, in MakeGraph()
61 #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()
91 #define BM_SPLITV_2D(split_dim, num_split, total_size0, total_size1) \ argument
|
| D | split_op.cc | 59 const int32 split_dim = in ComputeEasyCases() local
136 const TensorShape& input_shape, int32 split_dim, in operator ()()
217 const int32 split_dim = in Compute() local
287 const int32 split_dim = in Compute() local
344 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
|
| D | sparse_split_op.cc | 33 const int64 split_dim = context->input(0).scalar<int64>()(); 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
110 int64 split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims() in Compile() local
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local.py | 707 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 | 573 const int split_dim, const int num_split, in Split()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 801 split_dim=None): argument
|
| /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 | 581 int64 split_dim = c->Value(split_dimension); in __anondb9326b20a02() local
642 int64 split_dim = c->Value(split_dimension); in __anondb9326b20b02() local
|
| D | array_ops_test.cc | 1158 Tensor split_dim = test::AsTensor<int32>({1, 2}); in TEST() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1156 const int split_dim = split_dim_data[0]; in ConvertSplitOperator() local
1179 const int split_dim = split_dim_data[0]; in ConvertSplitVOperator() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 777 const int split_dim = input_shape.dims(axis); in ProcessTensorFlowSplitOperator() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 1752 auto split_dim = ops::Const(scope.WithOpName("split_dim"), {0}, {}); in TEST_F() local
1796 auto split_dim = ops::Const(scope.WithOpName("split_dim"), {0}, {}); in TEST_F() local
|