Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
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.variant>, tensor<i32>) -> tensor<*xf32>
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_TensorListStack.pbtxt2 graph_op_name: "TensorListStack"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_TensorListStack.pbtxt2 graph_op_name: "TensorListStack"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTensorListStack.pbtxt2 name: "TensorListStack"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DTensorListStack.pbtxt2 name: "TensorListStack"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TensorListStack.pbtxt2 graph_op_name: "TensorListStack"
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlower-static-tensor-list.mlir13 …%1 = "tf.TensorListStack"(%0, %arg0) : (tensor<!tf.variant<tensor<3xi32>>>, tensor<1xi32>) -> tens…
21 // CHECK-NOT: tf.TensorListStack
22 …%1 = "tf.TensorListStack"(%0, %arg0) : (tensor<!tf.variant<tensor<3xi32>>>, tensor<1xi32>) -> tens…
29 …%2 = "tf.TensorListStack"(%0, %arg1) : (tensor<!tf.variant<tensor<10xf32>>>, tensor<1xi32>) -> ten…
40 …%2 = "tf.TensorListStack"(%0, %arg1) : (tensor<!tf.variant<tensor<*xf32>>>, tensor<1xi32>) -> tens…
51 …%1 = "tf.TensorListStack"(%0, %cst) {num_elements = 2 : i64} : (tensor<!tf.variant<tensor<3xf32>>>…
64 …%2 = "tf.TensorListStack"(%1, %arg1) : (tensor<!tf.variant<tensor<10xf32>>>, tensor<1xi32>) -> ten…
109 …%2 = "tf.TensorListStack"(%1, %arg1) : (tensor<!tf.variant<tensor<f32>>>, tensor<0xi32>) -> tensor…
200 …%2 = "tf.TensorListStack"(%1, %cst) : (tensor<!tf.variant<tensor<2xf32>>>, tensor<i32>) -> tensor<…
232 …%2 = "tf.TensorListStack"(%1, %arg1) : (tensor<!tf.variant<tensor<10xf32>>>, tensor<1xi32>) -> ten…
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.cu.cc44 TensorListStack<GPUDevice, T>) \
Dlist_kernels.h59 class TensorListStack : public OpKernel {
63 explicit TensorListStack(OpKernelConstruction* c) : OpKernel(c) { in TensorListStack() function
Dlist_kernels.cc558 TensorListStack<CPUDevice, T>) \
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md630 * `raw_ops.TensorListStack`
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtensor_list_ops_decomposition.mlir81 …%stack = "tf.TensorListStack"(%addn2, %elem_shape) : (tensor<!tf.variant<tensor<f32>>>, tensor<0xi…
156 …%stack = "tf.TensorListStack"(%scatter, %elem_shape) : (tensor<!tf.variant<tensor<8x9xf32>>>, tens…
Dshape_inference.mlir473 …// CHECK: tf.TensorListStack{{.*}}: (tensor<!tf.variant<tensor<2x2xf32>>>, tensor<i32>) -> tensor<…
474 …%8 = "tf.TensorListStack"(%6, %7) {device = "", num_elements = -1 : i64} : (tensor<*x!tf.variant>,…
492 …%stack = "tf.TensorListStack"(%tl_2, %elem_shape) {num_elements = -1 : i64} : (tensor<!tf.variant<…
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision_test.cc835 Output tl1r1 = ops::TensorListStack(s.WithOpName("tl1r1"), tl1.output_handle, in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md2996 ### tf.TensorListStack
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt5064 name: "TensorListStack"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt5064 name: "TensorListStack"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go16110 func TensorListStack(scope *Scope, input_handle tf.Output, element_shape tf.Output, element_dtype t… func
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td16473 def TF_TensorListStackOp : TF_Op<"TensorListStack", [NoSideEffect]> {
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt55857 name: "TensorListStack"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt83807 name: "TensorListStack"