Home
last modified time | relevance | path

Searched refs:TensorListSetItem (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
Dtensor-list.pbtxt25 name: "TensorListSetItem/index"
48 name: "TensorListSetItem/item"
169 name: "TensorListSetItem"
170 op: "TensorListSetItem"
172 input: "TensorListSetItem/index"
173 input: "TensorListSetItem/item"
186 input: "TensorListSetItem"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dtensor-list.pbtxt27 name: "TensorListSetItem/index"
50 name: "TensorListSetItem/item"
171 name: "TensorListSetItem"
172 op: "TensorListSetItem"
174 input: "TensorListSetItem/index"
175 input: "TensorListSetItem/item"
188 input: "TensorListSetItem"
217 # CHECK: tf.TensorListSetItem{{.*}}(tensor<!tf_type.variant<tensor<*xf32>>>, tensor<i32>, tensor<2…
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTensorListSetItem.pbtxt2 name: "TensorListSetItem"
25 name: "TensorListSetItem"
55 name: "TensorListSetItem"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TensorListSetItem.pbtxt2 graph_op_name: "TensorListSetItem"
Dapi_def_TensorSetItem.pbtxt2 graph_op_name: "TensorListSetItem"
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_TensorListSetItem.pbtxt2 graph_op_name: "TensorListSetItem"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_TensorListSetItem.pbtxt2 graph_op_name: "TensorListSetItem"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DTensorListSetItem.pbtxt2 name: "TensorListSetItem"
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.cc437 class TensorListSetItem : public OpKernel { class
439 explicit TensorListSetItem(OpKernelConstruction* c) : OpKernel(c) { in TensorListSetItem() function in tensorflow::TensorListSetItem
473 TensorListSetItem);
482 TensorListSetItem);
497 TensorListSetItem);
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlower-static-tensor-list-enable-dynamic-update-slice.mlir8 …%1 = "tf.TensorListSetItem"(%0, %arg2, %arg3) : (tensor<!tf_type.variant<tensor<10xf32>>>, tensor<…
Dlower-static-tensor-list.mlir75 …%1 = "tf.TensorListSetItem"(%0, %arg2, %arg3) : (tensor<!tf_type.variant<tensor<10xf32>>>, tensor<…
122 …%1 = "tf.TensorListSetItem"(%0, %arg2, %arg3) : (tensor<!tf_type.variant<tensor<f32>>>, tensor<i32…
220 …%1 = "tf.TensorListSetItem"(%0, %arg2, %arg3) : (tensor<!tf_type.variant<tensor<*xf32>>>, tensor<i…
577 …%0 = "tf.TensorListSetItem"(%arg0, %cst, %cst_0) : (tensor<!tf_type.variant<tensor<10xf32>>>, tens…
581 // CHECK-NOT: tf.TensorListSetItem
587 …%0 = "tf.TensorListSetItem"(%arg0, %cst, %cst_0) : (tensor<!tf_type.variant<tensor<10xf32>>>, tens…
591 // CHECK-NOT: tf.TensorListSetItem
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_liveness_analysis_test.cc506 …3]{1,0} reshape(f32[3]{0} %rng.21), metadata={op_name="while/TensorArrayV2Write/TensorListSetItem"} in TEST_F()
508 …le-element.18, s32[] %constant.12), metadata={op_name="while/TensorArrayV2Write/TensorListSetItem"} in TEST_F()
/external/tensorflow/tensorflow/core/ir/utils/
Dshape_inference_utils_test.cc56 …9 = TensorListSetItem(%TensorListReserve, %Const_7, %Const_8) name("TensorListSetItem") {element_d…
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision_test.cc761 ops::TensorListSetItem(s.WithOpName("tl1w1"), tl1.handle, idx1, input); in TEST_F()
764 ops::TensorListSetItem(s.WithOpName("tl1w2"), tl1.handle, idx2, allow1); in TEST_F()
774 ops::TensorListSetItem(s.WithOpName("tl1w3"), tl1.handle, idx3, allow2); in TEST_F()
781 ops::TensorListSetItem(s.WithOpName("tl2w1"), tl2.handle, idx1, input); in TEST_F()
1560 ops::TensorListSetItem(s.WithOpName("tl1w1"), tl1.handle, idx1, input); in TEST_F()
1563 ops::TensorListSetItem(s.WithOpName("tl1w2"), tl1.handle, idx2, allow1); in TEST_F()
1573 ops::TensorListSetItem(s.WithOpName("tl1w3"), tl1.handle, idx3, allow2); in TEST_F()
1580 ops::TensorListSetItem(s.WithOpName("tl2w1"), tl2.handle, idx1, input); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dshape_inference.mlir479 …// CHECK: tf.TensorListSetItem{{.*}}: (tensor<!tf_type.variant<tensor<2x2xf32>>>, tensor<i32>, ten…
480 …%6 = "tf.TensorListSetItem"(%3, %4, %5) {device = ""} : (tensor<!tf_type.variant<tensor<*xf32>>>, …
561 …%tl_3 = "tf.TensorListSetItem"(%tl_1, %zero, %elem_1) : (tensor<!tf_type.variant<tensor<?x1xf32>>>…
577 …%tl_3 = "tf.TensorListSetItem"(%tl_1, %zero, %elem_1) : (tensor<!tf_type.variant<tensor<?x1xf32>>>…
621 …%tl_loop = "tf.TensorListSetItem"(%barg1, %index, %elem) : (tensor<!tf_type.variant<tensor<?x1xf32…
640 …%tl_true = "tf.TensorListSetItem"(%tl_0, %zero, %elem_0) : (tensor<!tf_type.variant<tensor<?x1xf32…
643 …%tl_false = "tf.TensorListSetItem"(%tl_0, %zero, %elem_1) : (tensor<!tf_type.variant<tensor<?x1xf3…
646 …%tl_2 = "tf.TensorListSetItem"(%tl_1, %one, %elem_2) : (tensor<!tf_type.variant<tensor<?x1xf32>>>,…
667 …%tl_loop = "tf.TensorListSetItem"(%barg1, %index, %elem_0) : (tensor<!tf_type.variant<tensor<?x1xf…
671 …%tl_set_item = "tf.TensorListSetItem"(%while#1, %one, %elem_1) : (tensor<!tf_type.variant<tensor<?…
[all …]
Dtensor_list_ops_decomposition.mlir67 …%set = "tf.TensorListSetItem"(%tl, %arg0, %elem) : (tensor<!tf_type.variant<tensor<f32>>>, tensor<…
/external/tensorflow/tensorflow/dtensor/mlir/
Dspmd_expanders.cc503 REGISTER_SPMD(TensorListSetItem, TF::TensorListSetItemOp,
/external/tensorflow/tensorflow/lite/toco/
Dtoco_flags.proto294 // Enable converting to DynamicUpdateSlice op (for ops like TensorListSetItem)
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dpasses.td166 "bool", "false", "When specified to true, lower TensorListSetItem with "
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md684 * `raw_ops.TensorListSetItem`
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtf_passes.td1938 with decremented size. Each `tf.TensorListSetItem` will be turned into a
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md3179 ### tf.TensorListSetItem
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt5136 name: "TensorListSetItem"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt5136 name: "TensorListSetItem"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go51900 func TensorListSetItem(scope *Scope, input_handle tf.Output, index tf.Output, item tf.Output) (outp… func

12