Home
last modified time | relevance | path

Searched refs:SplitV (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_SplitV.pbtxt2 graph_op_name: "SplitV"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_SplitV.pbtxt2 graph_op_name: "SplitV"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DSplitV.pbtxt2 name: "SplitV"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SplitV.pbtxt2 graph_op_name: "SplitV"
/external/tensorflow/tensorflow/compiler/jit/tests/
Dopens2s_gnmt_mixed_precision.golden_summary200 SplitV 6
Dopens2s_gnmt_mixed_precision.pbtxt.gz
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dprepare-composite-functions-tf.mlir175 // CHECK: [[VAL_12:%.*]]:4 = "tf.SplitV"([[VAL_7]], [[VAL_10]], [[VAL_11]]) : (tensor<40x…
178 // CHECK: [[VAL_15:%.*]]:4 = "tf.SplitV"([[VAL_9]], [[VAL_13]], [[VAL_14]]) : (tensor<40x…
181 // CHECK: [[VAL_18:%.*]]:4 = "tf.SplitV"([[VAL_5]], [[VAL_16]], [[VAL_17]]) : (tensor<40x…
Dlegalize-tf.mlir958 …%0:2 = "tf.SplitV"(%arg0, %arg1, %arg2) : (tensor<1x4x3x3xf32>, tensor<2xi32>, tensor<i32>) -> (te…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf-ops.mlir1739 …%0:2 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<f32>, tensor<2xi32>, tensor<i32>) -…
1747 …%0:2 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<4x4xf32>, tensor<2xi32>, tensor<1xi…
1756 …%0:2 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<4x4xf32>, tensor<2xi32>, tensor<i32…
1764 …%0:2 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<4x4xf32>, tensor<2x2xi32>, tensor<i…
1774 …%0:2 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<4x4xf32>, tensor<2xi32>, tensor<i32…
1784 …%0:2 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<4x4xf32>, tensor<2xi32>, tensor<i32…
1794 …%0:2 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<4x4xf32>, tensor<2xi32>, tensor<i32…
1803 …%0:2 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<4x4xf32>, tensor<2xi32>, tensor<i32…
1810 …%0:2 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<4x4xf32>, tensor<2xi32>, tensor<i32…
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf.cc113 DECL_CONVERT_OP(SplitV);
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc1128 class SplitV class
1843 ops.push_back(MakeUnique<SplitV>(::tflite::BuiltinOperator_SPLIT_V, in BuildOperatorList()
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dcpu_supported_ops.md227 `SplitV` | `Tlen={int32,int64}`<br>`T={bool,complex64,double,float,int…
Dgpu_supported_ops.md224 `SplitV` | `Tlen={int32,int64}`<br>`T={bool,complex64,double,float,int…
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir2847 // tf.SplitV legalization
2858 …%0:3 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<4x6xf32>, tensor<3xi32>, tensor<i32…
2870 …%0:3 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<?x6xf32>, tensor<3xi32>, tensor<i32…
2881 …%0:3 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<4x6xf32>, tensor<3xi32>, tensor<i32…
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc394 HANDLE_OPERATORTYPENAME_CASE(SplitV) in OperatorTypeName()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding_test.cc2049 ops::SplitV s1(scope.WithOpName("s1"), in1, size_splits1, split_dim, 1); in TEST_F()
2050 ops::SplitV s2(scope.WithOpName("s2"), in2, size_splits2, split_dim, 2); in TEST_F()
Dgeneric_layout_optimizer_transposer_test.cc3205 ops::SplitV(scope.WithOpName("splitv").WithDevice("/device:GPU:0"), in TEST_F()
Darithmetic_optimizer_test.cc3256 ops::SplitV split2(s.WithOpName("split2"), x, size_splits2, axis, 2); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td2626 of the resultant sub-tensors is decided by `size-splits`. Same as tf.SplitV.
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt4168 name: "SplitV"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt4168 name: "SplitV"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td5987 def TF_SplitVOp : TF_Op<"SplitV", [NoSideEffect]> {
/external/tensorflow/tensorflow/go/op/
Dwrappers.go8753 func SplitV(scope *Scope, value tf.Output, size_splits tf.Output, axis tf.Output, num_split int64) … func
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt35093 name: "SplitV"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt45320 name: "SplitV"

12