Home
last modified time | relevance | path

Searched refs:split_v (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dsplit_v.cc26 namespace split_v { namespace
125 /*prepare=*/split_v::Prepare, in Register_SPLIT_V()
126 /*invoke=*/split_v::Eval, in Register_SPLIT_V()
DBUILD244 "split_v.cc",
/external/tensorflow/tensorflow/lite/kernels/
Dsplit_v.cc31 namespace split_v { namespace
220 static TfLiteRegistration r = {nullptr, nullptr, split_v::Prepare, in Register_SPLIT_V()
221 split_v::Eval}; in Register_SPLIT_V()
DAndroid.bp151 "split_v.cc",
DBUILD626 "split_v.cc",
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dops.mlir1741 …%0 = "tfl.split_v"(%arg0, %arg1, %arg2) {num_splits = 1 : i32} : (tensor<10x!quant.uniform<u8:f32,…
2088 …// expected-error @+1 {{'tfl.split_v' op attribute 'num_splits' failed to satisfy constraint: 32-b…
2089 …"tfl.split_v"(%arg0, %size_splits, %split_dim) {num_splits = 0 : i32} : (tensor<16xf32>, tensor<0x…
2098 // expected-error @+1 {{'tfl.split_v' op output count should match 'num_splits' attribute}}
2099 …%0, %1 = "tfl.split_v"(%arg0, %size_splits, %split_dim) {num_splits = 4 : i32} : (tensor<16xf32>, …
2108 …// expected-error @+1 {{'tfl.split_v' op operand #1 must be 1D tensor of 32-bit signless integer v…
2109 …%0 = "tfl.split_v"(%arg0, %size_splits, %split_dim) {num_splits = 1 : i32} : (tensor<16x4x4xf32>, …
2117 …// expected-error @+1 {{'tfl.split_v' op operand #1 must be 1D tensor of 32-bit signless integer v…
2118 …%0 = "tfl.split_v"(%arg0, %size_splits, %split_dim) {num_splits = 1 : i32} : (tensor<16x4x4xf32>, …
2127 …// expected-error @+1 {{'tfl.split_v' op elements of 'size_splits' should be greater than or equal…
[all …]
Dlegalize-tf.mlir1120 …// CHECK: "tfl.split_v"(%arg0, %arg1, %arg2) {num_splits = 2 : i32} : (tensor<1x4x3x3xf32>, tensor…
/external/tensorflow/tensorflow/lite/
Dtflite_static.bp134 "kernels/split_v.cc",
/external/rust/crates/bindgen/src/
Dlib.rs2531 let split_v: Option<Vec<&str>> = raw_v in clang_version() localVariable
2536 match split_v { in clang_version()
/external/tensorflow/tensorflow/lite/micro/tools/make/
DMakefile355 tensorflow/lite/micro/kernels/split_v.cc \
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py2145 return gen_array_ops.split_v(
/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py352 split = gen_array_ops.split_v(
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md4089 ### tfl.split_v
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td3296 def TFL_SplitVOp : TFL_Op<"split_v", [NoSideEffect, SameOperandsAndResultsScale]> {