| /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
59 #define BM_SPLIT_2D(split_dim, num_split, chunk_size0, chunk_size1) \ argument
|
| D | split_v_op_test.cc | 77 #define BM_SPLITV_1D(num_split, total_size) \ argument
92 #define BM_SPLITV_2D(split_dim, num_split, dim0, dim1) \ argument
109 #define BM_SPLITV_3D(split_dim, num_split, dim0, dim1, dim2) \ argument
|
| D | split_op.cc | 58 const int32_t num_split = num_outputs(); in ComputeEasyCases() local
138 const ReshapeResultType& reshape_result, int32_t num_split, in operator ()()
210 const int32_t num_split = Base::num_outputs(); in Compute() local
286 const int32_t num_split = Base::num_outputs(); in Compute() local
|
| D | split_v_op.cc | 55 const int32_t num_split = context->num_outputs(); in ComputeEasyCases() local
279 const int num_split = split_start_points.size(); in operator ()() local
357 const int32_t num_split = Base::num_outputs(); in Compute() local
436 const int32_t num_split = Base::num_outputs(); in Compute() local
|
| D | split_lib_gpu.cu.cc | 77 const int32 num_split = output_ptr_data.size; in SplitOpKernel() local
173 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_t num_split = num_outputs(); in Compile() local
104 const int32_t num_split = num_outputs(); in Compile() local
|
| /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/ |
| D | GraphOperationTest.java | 182 private static int split(int[] values, int num_split) { in split()
193 private static int splitWithInputList(int[] values, int num_split, String name) { in splitWithInputList()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 985 num_split=None, argument
1057 num_split=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 343 auto num_split = rewriter.getI32IntegerAttr(tf_split_op.num_split()); in matchAndRewrite() local
356 auto num_split = rewriter.getI32IntegerAttr(tf_splitv_op.num_split()); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 496 const int split_dim, const int num_split, in Split()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 49 mlir::LogicalResult CreateSplitOp(const int num_split, in CreateSplitOp()
|
| /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/framework/ |
| D | node_def_util_test.cc | 596 int64_t num_split = 1000000000000; in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 944 uint16_t num_split; member
981 uint16_t num_split; member
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 1182 const int num_split = GetIntAttr(node, "num_split"); in ConvertSplitOperator() local
1201 const int num_split = GetIntAttr(node, "num_split"); in ConvertSplitVOperator() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 578 int num_split = c->num_outputs(); in __anon5ce23fde0a02() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1550 int32_t num_split = -1; in matchAndRewrite() local
|
| D | legalize_common.cc | 1919 Value input_value, int32_t num_split, int32_t axis) { in convertSplitOp()
|
| D | legalize_tfl.cc | 1957 int32_t num_split = -1; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 6303 auto get_split_nodedef = [](DataType dtype, int num_split) -> NodeDef { in __anonbe2eabc32e02()
6319 int num_split; in TestConvertSplit() member
|
| D | convert_nodes.cc | 4661 const int num_split = attrs.get<int64>("num_split"); in ConvertSplit() local
|
| /external/armnn/src/armnnTfParser/ |
| D | TfParser.cpp | 2798 uint32_t num_split = ReadMandatoryNodeUint32Attribute(nodeDef, "num_split"); in ParseSplit() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 1540 const int num_split = node->attr().at("num_split").i(); in IsSupported() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 841 int num_split = sharding_it->second; in CreateOrGetSplitNodesForInputSharding() local
|