Home
last modified time | relevance | path

Searched refs:TensorListStack (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
H A Dtensor-list.pbtxt82 name: "TensorListStack/element_shape"
186 name: "TensorListStack"
187 op: "TensorListStack"
189 input: "TensorListStack/element_shape"
218 # CHECK: tf.TensorListStack{{.*}}(tensor<*x!tf_type.variant>, tensor<i32>) -> tensor<*xf32>
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
H A Dtensor-list.pbtxt80 name: "TensorListStack/element_shape"
184 name: "TensorListStack"
185 op: "TensorListStack"
187 input: "TensorListStack/element_shape"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
H A Dlower-static-tensor-list.mlir13 …%1 = "tf.TensorListStack"(%0, %arg0) : (tensor<!tf_type.variant<tensor<3xi32>>>, tensor<1xi32>) ->…
24 // CHECK-NOT: tf.TensorListStack
25 …%1 = "tf.TensorListStack"(%0, %arg0) : (tensor<!tf_type.variant<tensor<3xi32>>>, tensor<1xi32>) ->…
35 …%2 = "tf.TensorListStack"(%0, %arg1) : (tensor<!tf_type.variant<tensor<10xf32>>>, tensor<1xi32>) -…
48 …%2 = "tf.TensorListStack"(%0, %arg1) : (tensor<!tf_type.variant<tensor<*xf32>>>, tensor<1xi32>) ->…
61 …%1 = "tf.TensorListStack"(%0, %cst) {num_elements = 2 : i64} : (tensor<!tf_type.variant<tensor<3xf…
76 …%2 = "tf.TensorListStack"(%1, %arg1) : (tensor<!tf_type.variant<tensor<10xf32>>>, tensor<1xi32>) -…
123 …%2 = "tf.TensorListStack"(%1, %arg1) : (tensor<!tf_type.variant<tensor<f32>>>, tensor<0xi32>) -> t…
222 …%2 = "tf.TensorListStack"(%1, %cst) : (tensor<!tf_type.variant<tensor<2xf32>>>, tensor<i32>) -> te…
275 …%2 = "tf.TensorListStack"(%1, %arg1) : (tensor<!tf_type.variant<tensor<10xf32>>>, tensor<1xi32>) -…
[all …]
H A Dlower-static-tensor-list-enable-dynamic-update-slice.mlir9 …%2 = "tf.TensorListStack"(%1, %arg1) : (tensor<!tf_type.variant<tensor<10xf32>>>, tensor<1xi32>) -…
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/java_api/
H A Dapi_def_TensorListStack.pbtxt2 graph_op_name: "TensorListStack"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/python_api/
H A Dapi_def_TensorListStack.pbtxt2 graph_op_name: "TensorListStack"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DTensorListStack.pbtxt2 name: "TensorListStack"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DTensorListStack.pbtxt2 name: "TensorListStack"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_TensorListStack.pbtxt2 graph_op_name: "TensorListStack"
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dlist_kernels.cu.cc44 TensorListStack<GPUDevice, T>) \
H A Dlist_kernels.cc617 TensorListStack<CPUDevice, T>) \
714 TensorListStack<CPUDevice, T>) \
H A Dlist_kernels.h164 class TensorListStack : public OpKernel {
168 explicit TensorListStack(OpKernelConstruction* c) : OpKernel(c) { in TensorListStack() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A Dtensor_list_ops_decomposition.mlir81 …%stack = "tf.TensorListStack"(%addn2, %elem_shape) : (tensor<!tf_type.variant<tensor<f32>>>, tenso…
156 …%stack = "tf.TensorListStack"(%scatter, %elem_shape) : (tensor<!tf_type.variant<tensor<8x9xf32>>>,…
H A Dshape_inference.mlir483 …// CHECK: tf.TensorListStack{{.*}}: (tensor<!tf_type.variant<tensor<2x2xf32>>>, tensor<i32>) -> te…
484 …%9 = "tf.TensorListStack"(%8, %7) {device = "", num_elements = -1 : i64} : (tensor<*x!tf_type.vari…
502 …%stack = "tf.TensorListStack"(%tl_2, %elem_shape) {num_elements = -1 : i64} : (tensor<!tf_type.var…
/aosp_15_r20/external/tensorflow/tensorflow/lite/g3doc/guide/
H A Dop_select_allowlist.md686 * `raw_ops.TensorListStack`
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dauto_mixed_precision_test.cc892 Output tl1r1 = ops::TensorListStack(s.WithOpName("tl1r1"), tl1.output_handle, in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
H A Dlegalization.md3183 ### tf.TensorListStack
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.raw_ops.pbtxt5144 name: "TensorListStack"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.raw_ops.pbtxt5144 name: "TensorListStack"
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go51955 func TensorListStack(scope *Scope, input_handle tf.Output, element_shape tf.Output, element_dtype t… func
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt59750 name: "TensorListStack"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_generated_ops.td18501 def TF_TensorListStackOp : TF_Op<"TensorListStack", [NoSideEffect]> {
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/
H A Dops_history.v2.pbtxt83782 name: "TensorListStack"