| /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 | 78 #define BM_SPLITV_1D(num_split, total_size) \ argument
93 #define BM_SPLITV_2D(split_dim, num_split, dim0, dim1) \ argument
110 #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
238 const int num_split = split_start_points.size(); in operator ()() local
303 const int32_t num_split = Base::num_outputs(); in Compute() local
382 const int32_t num_split = Base::num_outputs(); in Compute() local
|
| D | sparse_split_op_gpu.cu.cc | 119 Index input_nnz, int num_split, in LaunchSparseSplitSliceIndexesKernel()
136 Index input_nnz, int num_split, in SparseSplitFindSliceEndsKernel()
147 Index input_nnz, int num_split, in LaunchSparseSplitFindSliceEndsKernel()
|
| 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
|
| D | sparse_split_op.cc | 69 void SparseSplitOpImpl(OpKernelContext* context, int num_split, in SparseSplitOpImpl()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | split_op.cc | 38 const int32_t num_split = num_outputs(); in Compile() local
105 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 | 981 num_split=None, argument
1053 num_split=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 294 auto num_split = rewriter.getI32IntegerAttr(tf_split_op.num_split()); in matchAndRewrite() local
307 auto num_split = rewriter.getI32IntegerAttr(tf_splitv_op.num_split()); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 497 const int split_dim, const int num_split, in Split()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 45 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/dtensor/mlir/ |
| D | spmd_expander_common.cc | 97 Status CreateSplitOp(const int num_split, const int split_dimension, in CreateSplitOp()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 960 uint16_t num_split; member
997 uint16_t num_split; member
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 1189 const int num_split = GetIntAttr(node, "num_split"); in ConvertSplitOperator() local
1208 const int num_split = GetIntAttr(node, "num_split"); in ConvertSplitVOperator() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 579 int num_split = c->num_outputs(); in __anon091a71b70a02() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1543 int32_t num_split = -1; in matchAndRewrite() local
|
| D | legalize_common.cc | 2011 Value input_value, int32_t num_split, int32_t axis) { in convertSplitOp()
|
| D | legalize_tfl.cc | 2156 int32_t num_split = -1; in matchAndRewrite() 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/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 7915 auto get_split_nodedef = [](DataType dtype, int num_split) -> NodeDef { in __anon82ef283c3b02()
7931 int num_split; in TestConvertSplit() member
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 839 int num_split = sharding_it->second; in CreateOrGetSplitNodesForInputSharding() local
|