Searched refs:split_v (Results 1 – 14 of 14) sorted by relevance
/external/tensorflow/tensorflow/lite/micro/kernels/ |
D | split_v.cc | 26 namespace split_v { namespace 125 /*prepare=*/split_v::Prepare, in Register_SPLIT_V() 126 /*invoke=*/split_v::Eval, in Register_SPLIT_V()
|
D | BUILD | 244 "split_v.cc",
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | split_v.cc | 31 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()
|
D | Android.bp | 151 "split_v.cc",
|
D | BUILD | 626 "split_v.cc",
|
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
D | ops.mlir | 1741 …%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 …]
|
D | legalize-tf.mlir | 1120 …// CHECK: "tfl.split_v"(%arg0, %arg1, %arg2) {num_splits = 2 : i32} : (tensor<1x4x3x3xf32>, tensor…
|
/external/tensorflow/tensorflow/lite/ |
D | tflite_static.bp | 134 "kernels/split_v.cc",
|
/external/rust/crates/bindgen/src/ |
D | lib.rs | 2531 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/ |
D | Makefile | 355 tensorflow/lite/micro/kernels/split_v.cc \
|
/external/tensorflow/tensorflow/python/ops/ |
D | array_ops.py | 2145 return gen_array_ops.split_v(
|
/external/tensorflow/tensorflow/python/grappler/ |
D | layout_optimizer_test.py | 352 split = gen_array_ops.split_v(
|
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
D | legalization.md | 4089 ### tfl.split_v
|
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
D | tfl_ops.td | 3296 def TFL_SplitVOp : TFL_Op<"split_v", [NoSideEffect, SameOperandsAndResultsScale]> {
|