/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
H A D | tensor-list.pbtxt | 82 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 D | tensor-list.pbtxt | 80 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 D | lower-static-tensor-list.mlir | 13 …%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 D | lower-static-tensor-list-enable-dynamic-update-slice.mlir | 9 …%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 D | api_def_TensorListStack.pbtxt | 2 graph_op_name: "TensorListStack"
|
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/python_api/ |
H A D | api_def_TensorListStack.pbtxt | 2 graph_op_name: "TensorListStack"
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
H A D | TensorListStack.pbtxt | 2 name: "TensorListStack"
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
H A D | TensorListStack.pbtxt | 2 name: "TensorListStack"
|
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/ |
H A D | api_def_TensorListStack.pbtxt | 2 graph_op_name: "TensorListStack"
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | list_kernels.cu.cc | 44 TensorListStack<GPUDevice, T>) \
|
H A D | list_kernels.cc | 617 TensorListStack<CPUDevice, T>) \ 714 TensorListStack<CPUDevice, T>) \
|
H A D | list_kernels.h | 164 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 D | tensor_list_ops_decomposition.mlir | 81 …%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 D | shape_inference.mlir | 483 …// 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 D | op_select_allowlist.md | 686 * `raw_ops.TensorListStack`
|
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
H A D | auto_mixed_precision_test.cc | 892 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 D | legalization.md | 3183 ### tf.TensorListStack
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
H A D | tensorflow.raw_ops.pbtxt | 5144 name: "TensorListStack"
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
H A D | tensorflow.raw_ops.pbtxt | 5144 name: "TensorListStack"
|
/aosp_15_r20/external/tensorflow/tensorflow/go/op/ |
H A D | wrappers.go | 51955 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 D | ops.pbtxt | 59750 name: "TensorListStack"
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
H A D | tf_generated_ops.td | 18501 def TF_TensorListStackOp : TF_Op<"TensorListStack", [NoSideEffect]> {
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ |
H A D | ops_history.v2.pbtxt | 83782 name: "TensorListStack"
|