| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_op_test.cc | 29 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
|
| D | split_v_op_test.cc | 75 #define BM_SPLITV_1D(num_split, total_size) \ argument
91 #define BM_SPLITV_2D(split_dim, num_split, total_size0, total_size1) \ argument
|
| D | split_op.cc | 61 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
|
| D | split_v_op.cc | 55 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
|
| D | split_lib_gpu.cu.cc | 81 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/ |
| D | split_op.cc | 37 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/ |
| D | OperationTest.java | 169 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/ |
| D | sparse_ops.py | 798 num_split=None, argument
868 num_split=None, argument
|
| /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
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 573 const int split_dim, const int num_split, in Split()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 976 uint16 num_split; member
1013 uint16 num_split; member
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 931 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/ |
| D | types.h | 931 uint16 num_split; member
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 572 int num_split = c->num_outputs(); in __anon7c94107b0a02() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 2475 int num_split; in CopyAttrsSplit() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 1436 const int num_split = node->attr().at("num_split").i(); in IsSupported() local
|