Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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
59 #define BM_SPLIT_2D(split_dim, num_split, chunk_size0, chunk_size1) \ argument
H A Dsplit_v_op_test.cc78 #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
H A Dsplit_op.cc58 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
H A Dsplit_v_op.cc55 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
H A Dsparse_split_op_gpu.cu.cc119 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()
H A Dsplit_lib_gpu.cu.cc77 const int32 num_split = output_ptr_data.size; in SplitOpKernel() local
173 const int32 num_split = output_ptr_data.size; in SplitVOpKernel_fixed() local
H A Dsparse_split_op.cc69 void SparseSplitOpImpl(OpKernelContext* context, int num_split, in SparseSplitOpImpl()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dsplit_op.cc38 const int32_t num_split = num_outputs(); in Compile() local
105 const int32_t num_split = num_outputs(); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
H A DGraphOperationTest.java182 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()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dsparse_ops.py981 num_split=None, argument
1053 num_split=None, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dlegalize_tf.cc294 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
/aosp_15_r20/external/tensorflow/tensorflow/core/util/sparse/
H A Dsparse_tensor.h497 const int split_dim, const int num_split, in Split()
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/gallivm/
H A Dlp_bld_pack.c650 int num_split = src_type.width * src_type.length / 128; in lp_build_pack2() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dxla_sharding_util.cc45 mlir::LogicalResult CreateSplitOp(const int num_split, in CreateSplitOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dnode_def_util_test.cc596 int64_t num_split = 1000000000000; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dspmd_expander_common.cc97 Status CreateSplitOp(const int num_split, const int split_dimension, in CreateSplitOp()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dtypes.h960 uint16_t num_split; member
997 uint16_t num_split; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dimport_tensorflow.cc1189 const int num_split = GetIntAttr(node, "num_split"); in ConvertSplitOperator() local
1208 const int num_split = GetIntAttr(node, "num_split"); in ConvertSplitVOperator() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Darray_ops.cc579 int num_split = c->num_outputs(); in __anon38bbb0e80a02() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_tf.cc1543 int32_t num_split = -1; in matchAndRewrite() local
H A Dlegalize_common.cc2011 Value input_value, int32_t num_split, int32_t axis) { in convertSplitOp()
H A Dlegalize_tfl.cc2156 int32_t num_split = -1; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer.cc1540 const int num_split = node->attr().at("num_split").i(); in IsSupported() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc7915 auto get_split_nodedef = [](DataType dtype, int num_split) -> NodeDef { in __anond617510d3b02()
7931 int num_split; in TestConvertSplit() member
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc839 int num_split = sharding_it->second; in CreateOrGetSplitNodesForInputSharding() local

12