/external/tensorflow/tensorflow/core/kernels/ |
D | split_op_test.cc | 29 static Graph* MakeGraph(int split_dim, int num_split, in MakeGraph() argument 33 in_shape.set_dim(split_dim, in_shape.dim_size(split_dim) * num_split); in MakeGraph() 41 .Attr("num_split", num_split) in MakeGraph() 46 #define BM_SPLIT_1D(num_split, chunk_size) \ argument 47 static void BM_Split_1d_##num_split##_##chunk_size(int iters) { \ 49 testing::ItemsProcessed(static_cast<int64>(iters) * num_split * \ 52 strings::Printf("1-D %d chunks of %d each", num_split, chunk_size); \ 55 auto g = MakeGraph(/* split_dim = */ 0, num_split, {chunk_size}); \ 59 BENCHMARK(BM_Split_1d_##num_split##_##chunk_size); 61 #define BM_SPLIT_2D(split_dim, num_split, chunk_size0, chunk_size1) \ argument [all …]
|
D | split_op.cc | 61 const int32 num_split = num_outputs(); in ComputeEasyCases() local 70 context, num_split > 0, in ComputeEasyCases() 72 "Number of ways to split should be > 0, but got ", num_split)); in ComputeEasyCases() 74 OP_REQUIRES(context, input_shape.dim_size(split_dim) % num_split == 0, in ComputeEasyCases() 79 ") ", "and num_split ", num_split)); in ComputeEasyCases() 81 if (num_split == 1) { in ComputeEasyCases() 98 const int64 delta = input_shape.dim_size(0) / num_split; in ComputeEasyCases() 99 for (int i = 0; i < num_split; ++i) { in ComputeEasyCases() 141 const ReshapeResultType& reshape_result, int32 num_split, in operator ()() argument 148 (num_split >= 4 && in operator ()() [all …]
|
D | split_v_op.cc | 55 const int32 num_split = context->num_outputs(); in ComputeEasyCases() local 71 split_tensor.dims() == 1 && split_tensor.NumElements() == num_split, in ComputeEasyCases() 85 context, num_split > 0, in ComputeEasyCases() 87 "Number of ways to split should be > 0, but got ", num_split)); in ComputeEasyCases() 98 if (num_split == 1) { in ComputeEasyCases() 151 for (int i = 0; i < num_split; ++i) { in ComputeEasyCases() 206 const int num_split = split_start_points.size(); in operator ()() local 208 (num_split >= 4 && in operator ()() 209 input_element_count >= std::max(num_threads, num_split) * 4096 && in operator ()() 210 input_element_count < num_split * 180 * 1024); in operator ()() [all …]
|
D | split_v_op_test.cc | 75 #define BM_SPLITV_1D(num_split, total_size) \ argument 76 static void BM_SplitV_1d_##num_split##_##total_size(int iters) { \ 80 strings::Printf("1-D %d chunks totaling %d", num_split, total_size); \ 84 GenerateRandomIntsWithSum(total_size, num_split), \ 89 BENCHMARK(BM_SplitV_1d_##num_split##_##total_size); 91 #define BM_SPLITV_2D(split_dim, num_split, total_size0, total_size1) \ argument 93 BM_SplitV_2d_##split_dim##_##num_split##_##total_size0##_##total_size1( \ 101 num_split, split_dim, total_size0, total_size1); \ 106 GenerateRandomIntsWithSum(total_size_vec[split_dim], num_split), \ 112 BM_SplitV_2d_##split_dim##_##num_split##_##total_size0##_##total_size1);
|
D | split_lib_gpu.cu.cc | 81 const int32 num_split = output_ptr_data.size; in SplitOpKernel() local 86 eigen_assert(split_dim_size % num_split == 0); in SplitOpKernel() 89 int32 piece_size = split_dim_size / num_split; in SplitOpKernel() 177 const int32 num_split = output_ptr_data.size; in SplitVOpKernel_fixed() local 184 int32 piece_size = suffix_dim_size / num_split; in SplitVOpKernel_fixed()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | split_op.cc | 37 const int32 num_split = num_outputs(); in Compile() local 57 ctx, num_split > 0, in Compile() 59 "Number of ways to split should be > 0, but got ", num_split)); in Compile() 62 ctx, input_shape.dim_size(split_dim) % num_split == 0, in Compile() 67 "and num_split ", num_split)); in Compile() 71 const int32 slice_size = input_shape.dim_size(split_dim) / num_split; in Compile() 88 for (int i = 0; i < num_split; ++i) { in Compile() 104 const int32 num_split = num_outputs(); in Compile() local 124 ctx, num_split > 0, in Compile() 126 "Number of ways to split should be > 0, but got ", num_split)); in Compile() [all …]
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
D | SparseSplit.pbtxt | 22 number_attr: "num_split" 27 number_attr: "num_split" 32 number_attr: "num_split" 35 name: "num_split"
|
D | Split.pbtxt | 14 number_attr: "num_split" 17 name: "num_split"
|
D | SplitV.pbtxt | 18 number_attr: "num_split" 21 name: "num_split"
|
/external/tensorflow/tensorflow/core/api_def/base_api/ |
D | api_def_SparseSplit.pbtxt | 45 name: "num_split" 50 summary: "Split a `SparseTensor` into `num_split` tensors along one dimension." 52 If the `shape[split_dim]` is not an integer multiple of `num_split`. Slices 53 `[0 : shape[split_dim] % num_split]` gets one extra dimension. 54 For example, if `split_dim = 1` and `num_split = 2` and the input is
|
D | api_def_Split.pbtxt | 22 `values.shape[split_dim] / num_split`. 26 name: "num_split" 32 summary: "Splits a tensor into `num_split` tensors along one dimension."
|
D | api_def_SplitV.pbtxt | 33 summary: "Splits a tensor into `num_split` tensors along one dimension."
|
/external/tensorflow/tensorflow/core/util/sparse/ |
D | sparse_tensor.h | 173 const int num_split, std::vector<SparseTensor>* result); 459 const int split_dim, const int num_split, in Split() argument 464 output_indices.reserve(num_split); in Split() 465 output_values.reserve(num_split); in Split() 466 output_shapes.reserve(num_split); in Split() 470 output_indices_t.reserve(num_split); in Split() 471 output_values_t.reserve(num_split); in Split() 475 std::vector<int> num_values(num_split, 0); in Split() 478 const int split_size = split_dim_size / num_split; in Split() 480 if (!(num_split > 0 && num_split <= split_dim_size)) { in Split() [all …]
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | sparse_split_op_test.py | 83 sp_input=self._SparseTensor_4x6(), num_split=2, axis=0) 101 sp_input=self._SparseTensor_5x7(), num_split=3, axis=1) 119 sp_input=self._SparseTensor_5x7(), num_split=4, axis=1) 141 sp_input=self._SparseTensor_5x7(), num_split=2, axis=0) 156 sp_input=self._SparseTensor_5x7(), num_split=3, axis=0) 178 sp_input=self._SparseTensor_4x6(), num_split=4, axis=0) 201 sp_input=self._SparseTensor_4x6(), num_split=3, axis=1) 220 sp_input=self._SparseTensor_4x6(), num_split=6, axis=1) 250 sp_input=sp_input, num_split=2, axis=1) 264 sparse_ops.sparse_split(num_split=2, sp_input=1)
|
D | split_op_test.py | 132 num_split = np.random.randint(16, 25) 134 num_split = np.random.randint(2, 8) 135 size_splits = np.random.randint(2, 8, num_split, dtype=np.int32) 142 for i in range(num_split): 159 num_split = 1000 160 size_splits = np.random.randint(1, 3, num_split, dtype=np.int32) 168 for i in range(num_split): 290 num_split = np.random.randint(9, 15) 292 num_split = np.random.randint(2, 8) 293 shape[split_dim] = np.random.randint(2, 5) * num_split [all …]
|
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/ |
D | GraphOperationTest.java | 182 private static int split(int[] values, int num_split) { in split() argument 187 .setAttr("num_split", num_split) in split() 193 private static int splitWithInputList(int[] values, int num_split, String name) { in splitWithInputList() argument 198 .setAttr("num_split", num_split) in splitWithInputList()
|
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
D | legalize_tf.cc | 228 auto num_split = rewriter.getI32IntegerAttr(tf_split_op.num_split()); in matchAndRewrite() local 232 tf_split_op.value(), num_split); in matchAndRewrite() 243 auto num_split = rewriter.getI32IntegerAttr(tf_splitv_op.num_split()); in matchAndRewrite() local 247 tf_splitv_op.split_dim(), num_split); in matchAndRewrite()
|
/external/mesa3d/src/gallium/auxiliary/gallivm/ |
D | lp_bld_pack.c | 650 int num_split = src_type.width * src_type.length / 128; in lp_build_pack2() local 662 assert(num_split <= LP_MAX_VECTOR_WIDTH / 128); in lp_build_pack2() 664 for (i = 0; i < num_split / 2; i++) { in lp_build_pack2() 675 for (i = 0; i < num_split / 2; i++) { in lp_build_pack2() 680 tmpres[i+num_split/2] = lp_build_intrinsic_binary(builder, intrinsic, in lp_build_pack2() 684 tmpres[i+num_split/2] = LLVMBuildBitCast(builder, tmpres[i+num_split/2], in lp_build_pack2() 688 res = lp_build_concat(gallivm, tmpres, ndst_type, num_split); in lp_build_pack2()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
D | derived_size_attr.mlir | 5 // CHECK: key: "num_split"
|
/external/tensorflow/tensorflow/python/ops/ |
D | sparse_ops.py | 849 num_split=None, argument 893 if num_split is None: 907 num_split, 910 for i in range(0, num_split): 919 num_split=None, argument 956 num_split=num_split,
|
/external/tensorflow/tensorflow/lite/toco/tflite/ |
D | operator_test.cc | 333 op.num_split = 123; in TEST_F() 336 EXPECT_EQ(op.num_split, output_toco_op->num_split); in TEST_F() 341 op.num_split = 123; in TEST_F() 344 EXPECT_EQ(op.num_split, output_toco_op->num_split); in TEST_F()
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | unpack.cc | 87 op_params.num_split = output_count; in UnpackImpl()
|
D | split.cc | 129 op_params.num_split = NumOutputs(node); \ in Eval()
|
D | split_v.cc | 169 op_params.num_split = NumOutputs(node); \ in Eval()
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.sparse.pbtxt | 101 …argspec: "args=[\'sp_input\', \'num_split\', \'axis\', \'name\'], varargs=None, keywords=None, def…
|