| /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 | 75 #define BM_SPLITV_1D(num_split, total_size) \ argument
90 #define BM_SPLITV_2D(split_dim, num_split, total_size0, total_size1) \ argument
|
| D | split_op.cc | 58 const int32 num_split = num_outputs(); in ComputeEasyCases() local
138 const ReshapeResultType& reshape_result, int32 num_split, in operator ()()
210 const int32 num_split = Base::num_outputs(); in Compute() local
286 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 | 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 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 | 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/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 307 auto num_split = rewriter.getI32IntegerAttr(tf_split_op.num_split()); in matchAndRewrite() local
320 auto num_split = rewriter.getI32IntegerAttr(tf_splitv_op.num_split()); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 988 num_split=None, argument
1060 num_split=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 49 mlir::LogicalResult CreateSplitOp(const int num_split, in CreateSplitOp()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 496 const int split_dim, const int num_split, in Split()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util_test.cc | 456 int64 num_split = 1000000000000; in TEST() local
|
| /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/lite/kernels/internal/ |
| D | types.h | 1112 uint16_t num_split; member
1149 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/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1544 int32_t num_split = -1; in matchAndRewrite() local
|
| D | legalize_common.cc | 1844 Value input_value, int32_t num_split, in convertSplitOp()
|
| D | legalize_tfl.cc | 2053 int32_t num_split = -1; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 578 int num_split = c->num_outputs(); in __anondb9326b20a02() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 5855 auto get_split_nodedef = [](DataType dtype, int num_split) -> NodeDef { in TEST_P()
5871 int num_split; in TEST_P() member
|
| D | convert_nodes.cc | 5032 const int num_split = attrs.get<int64>("num_split"); in ConvertSplit() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 1478 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 | 828 int num_split = sharding_it->second; in CreateOrGetSplitNodesForInputSharding() local
|