Home
last modified time | relevance | path

Searched refs:num_split (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Dsplit_op_test.cc29 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 …]
Dsplit_op.cc61 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 …]
Dsplit_v_op.cc55 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 …]
Dsplit_v_op_test.cc75 #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);
Dsplit_lib_gpu.cu.cc81 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/
Dsplit_op.cc37 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/
DSparseSplit.pbtxt22 number_attr: "num_split"
27 number_attr: "num_split"
32 number_attr: "num_split"
35 name: "num_split"
DSplit.pbtxt14 number_attr: "num_split"
17 name: "num_split"
DSplitV.pbtxt18 number_attr: "num_split"
21 name: "num_split"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseSplit.pbtxt45 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
Dapi_def_Split.pbtxt22 `values.shape[split_dim] / num_split`.
26 name: "num_split"
32 summary: "Splits a tensor into `num_split` tensors along one dimension."
Dapi_def_SplitV.pbtxt33 summary: "Splits a tensor into `num_split` tensors along one dimension."
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h173 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/
Dsparse_split_op_test.py83 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)
Dsplit_op_test.py132 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/
DGraphOperationTest.java182 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/
Dlegalize_tf.cc228 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/
Dlp_bld_pack.c650 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/
Dderived_size_attr.mlir5 // CHECK: key: "num_split"
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py849 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/
Doperator_test.cc333 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/
Dunpack.cc87 op_params.num_split = output_count; in UnpackImpl()
Dsplit.cc129 op_params.num_split = NumOutputs(node); \ in Eval()
Dsplit_v.cc169 op_params.num_split = NumOutputs(node); \ in Eval()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.sparse.pbtxt101 …argspec: "args=[\'sp_input\', \'num_split\', \'axis\', \'name\'], varargs=None, keywords=None, def…

123