Home
last modified time | relevance | path

Searched refs:TensorListReserve (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dtensor-list.pbtxt4 name: "TensorListReserve/num_elements"
105 name: "TensorListReserve/element_shape"
131 name: "TensorListReserve"
132 op: "TensorListReserve"
133 input: "TensorListReserve/element_shape"
134 input: "TensorListReserve/num_elements"
152 op: "TensorListReserve"
153 input: "TensorListReserve/element_shape"
154 input: "TensorListReserve/num_elements"
173 input: "TensorListReserve"
[all …]
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_TensorListReserve.pbtxt2 graph_op_name: "TensorListReserve"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_TensorListReserve.pbtxt2 graph_op_name: "TensorListReserve"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DTensorListReserve.pbtxt2 name: "TensorListReserve"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTensorListReserve.pbtxt2 name: "TensorListReserve"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TensorListReserve.pbtxt2 graph_op_name: "TensorListReserve"
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.cc295 class TensorListReserve : public OpKernel { class
297 explicit TensorListReserve(OpKernelConstruction* c) : OpKernel(c) { in TensorListReserve() function in tensorflow::TensorListReserve
321 TensorListReserve);
329 TensorListReserve);
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dshape_inference.mlir464 // CHECK: TensorListReserve{{.*}}-> tensor<!tf.variant<tensor<2x2x!tf.variant>>>
465 …%2 = "tf.TensorListReserve"(%0, %1) {device = ""} : (tensor<2xi32>, tensor<i32>) -> tensor<!tf.var…
466 // CHECK: TensorListReserve{{.*}}-> tensor<!tf.variant<tensor<2x2xf32>>>
467 …%3 = "tf.TensorListReserve"(%0, %1) {device = ""} : (tensor<2xi32>, tensor<i32>) -> tensor<!tf.var…
577 // CHECK: TensorListReserve
579 …%tl_0 = "tf.TensorListReserve"(%elem_shape, %size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf.va…
600 // CHECK: TensorListReserve
602 …%tl = "tf.TensorListReserve"(%elem_shape, %size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf.vari…
623 // CHECK: TensorListReserve
625 …%tl_0 = "tf.TensorListReserve"(%elem_shape, %size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf.va…
[all …]
Dtensor_list_ops_decomposition.mlir59 …%tl = "tf.TensorListReserve"(%elem_shape, %num) : (tensor<0xi32>, tensor<i32>) -> tensor<!tf.varia…
562 // CHECK-NOT: tf.TensorListReserve
565 …%tl1 = "tf.TensorListReserve"(%arg0, %max_size) : (tensor<*xi32>, tensor<i32>) -> tensor<!tf.varia…
611 …%tl = "tf.TensorListReserve"(%elem_shape, %num) : (tensor<0xi32>, tensor<i32>) -> tensor<!tf.varia…
Dmark_ops_for_outside_compilation.mlir340 // CHECK: "tf.TensorListReserve"
344 …%0 = "tf.TensorListReserve"(%arg0, %arg1) : (tensor<i32>, tensor<i32>) -> tensor<!tf.variant<tenso…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlower-static-tensor-list.mlir152 …%0 = "tf.TensorListReserve"(%arg0, %arg1) : (tensor<3xi32>, tensor<i32>) -> tensor<!tf.variant<ten…
168 …%0 = "tf.TensorListReserve"(%arg0, %arg1) : (tensor<?xi32>, tensor<i32>) -> tensor<!tf.variant<ten…
180 …%0 = "tf.TensorListReserve"(%cst, %arg0) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf.variant<tens…
197 …%0 = "tf.TensorListReserve"(%arg0, %arg1) : (tensor<i32>, tensor<i32>) -> tensor<!tf.variant<tenso…
441 …%0 = "tf.TensorListReserve"(%arg0, %arg1) : (tensor<i32>, tensor<i32>) -> tensor<!tf.variant<tenso…
445 // CHECK: %0 = "tf.TensorListReserve"(%arg0, %arg1) : (tensor<i32>, tensor<i32>) -> tensor<!tf.vari…
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-with-tf2xla.mlir24 // CHECK: tf.TensorListReserve
25 …%0 = "tf.TensorListReserve"(%arg0, %arg1) : (tensor<3xi32>, tensor<i32>) -> tensor<!tf.variant<ten…
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision_test.cc698 auto tl1 = ops::TensorListReserve(s.WithOpName("tl1"), {32, 32}, 8, DT_FLOAT); in TEST_F()
722 auto tl2 = ops::TensorListReserve(s.WithOpName("tl2"), shape, 8, DT_FLOAT); in TEST_F()
1284 auto tl1 = ops::TensorListReserve(s.WithOpName("tl1"), {32, 32}, 8, DT_FLOAT); in TEST_F()
1308 auto tl2 = ops::TensorListReserve(s.WithOpName("tl2"), shape, 8, DT_FLOAT); in TEST_F()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md623 * `raw_ops.TensorListReserve`
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test.cc1294 Output tensor_list_reserve = ops::TensorListReserve( in TEST()
1342 Output tensor_list_reserve = ops::TensorListReserve( in TEST()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.td802 def TF_TensorListReserveOp : TF_TensorListInitOp<"TensorListReserve"> {
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md2984 ### tf.TensorListReserve
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt5036 name: "TensorListReserve"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt5036 name: "TensorListReserve"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go26225 func TensorListReserve(scope *Scope, element_shape tf.Output, num_elements tf.Output, element_dtype… func
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt55664 name: "TensorListReserve"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt83613 name: "TensorListReserve"