- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for StackPopV2 (0.51 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/stack_ops_decomposition.mlir
func.func @if_else(%arg0: tensor<!tf_type.resource>) -> tensor<!tf_type.resource> { // CHECK-NOT: "tf.StackPopV2" // CHECK: "tf.Slice" // CHECK: "tf.AssignVariableOp"(%[[EARG1:.*]], // CHECK-NOT: "tf.StackPopV2" %pop = "tf.StackPopV2"(%arg0) : (tensor<!tf_type.resource>) -> tensor<f32> func.return %arg0 : tensor<!tf_type.resource> } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 25.8K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
// of this concession it is unsound to auto-cluster them because then // we'd create clusters we could not compile (because we can't constant // fold, say, a TensorArrayRead or a StackPopV2). But we don't // auto-cluster these operations today so we're good for now. const XlaResourceOpInfo* op_info = GetResourceOpInfoForOp(node->type_string());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
let arguments = (ins Arg<TF_ResourceTensor, [{The handle to a stack.}], [TF_StackFree]>:$handle ); let results = (outs); } def TF_StackPopV2Op : TF_Op<"StackPopV2", []> { let summary = "Pop the element at the top of the stack."; let arguments = (ins Arg<TF_ResourceTensor, [{The handle to a stack.}], [TF_StackRead, TF_StackWrite]>:$handle );
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)