Home
last modified time | relevance | path

Searched defs:num_split (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsplit_op_test.cc29 static Graph* MakeGraph(int split_dim, int num_split, in MakeGraph()
46 #define BM_SPLIT_1D(num_split, chunk_size) \ argument
61 #define BM_SPLIT_2D(split_dim, num_split, chunk_size0, chunk_size1) \ argument
Dsplit_v_op_test.cc75 #define BM_SPLITV_1D(num_split, total_size) \ argument
91 #define BM_SPLITV_2D(split_dim, num_split, total_size0, total_size1) \ argument
Dsplit_op.cc61 const int32 num_split = num_outputs(); in ComputeEasyCases() local
141 const ReshapeResultType& reshape_result, int32 num_split, in operator ()()
213 const int32 num_split = Base::num_outputs(); in Compute() local
289 const int32 num_split = Base::num_outputs(); in Compute() local
346 const int32 num_split = Base::num_outputs(); in Compute() local
Dsplit_v_op.cc55 const int32 num_split = context->num_outputs(); in ComputeEasyCases() local
206 const int num_split = split_start_points.size(); in operator ()() local
268 const int32 num_split = Base::num_outputs(); in Compute() local
347 const int32 num_split = Base::num_outputs(); in Compute() local
Dsplit_lib_gpu.cu.cc81 const int32 num_split = output_ptr_data.size; in SplitOpKernel() local
176 const int32 num_split = output_ptr_data.size; in SplitVOpKernel_fixed() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsplit_op.cc37 const int32 num_split = num_outputs(); in Compile() local
104 const int32 num_split = num_outputs(); in Compile() local
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
DOperationTest.java169 private static int split(int[] values, int num_split) { in split()
180 private static int splitWithInputList(int[] values, int num_split, String name) { in splitWithInputList()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py798 num_split=None, argument
868 num_split=None, argument
/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
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h573 const int split_dim, const int num_split, in Split()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h976 uint16 num_split; member
1013 uint16 num_split; member
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc931 const int num_split = GetIntAttr(node, "num_split"); in ConvertSplitOperator() local
950 const int num_split = GetIntAttr(node, "num_split"); in ConvertSplitVOperator() local
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h931 uint16 num_split; member
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc572 int num_split = c->num_outputs(); in __anon7c94107b0a02() local
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc2475 int num_split; in CopyAttrsSplit() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc1436 const int num_split = node->attr().at("num_split").i(); in IsSupported() local