Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TensorArrayCloseV3 (0.27 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/tensor_array_ops_decomposition.mlir

      // CHECK-SAME: -> tensor<3xf32>
      %read = "tf.TensorArrayReadV3"(%ta#0, %index, %write) : (tensor<!tf_type.resource>, tensor<i32>, tensor<f32>) -> tensor<3xf32>
      // CHECK-NOT: TensorArrayCloseV3
      "tf.TensorArrayCloseV3"(%ta#0) : (tensor<!tf_type.resource>) -> ()
      // CHECK: return %[[READ]] : tensor<3xf32>
      func.return %read: tensor<3xf32>
    }
    
    // -----
    
    // Test inferring shape from the first write.
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 49K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass.cc

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        static bool isCompatibleReturnTypes(TypeRange inferred, TypeRange actual) {
          return ArraysAreCastCompatible(inferred, actual);
        }
      }];
    }
    
    def TF_TensorArrayCloseV3Op : TF_Op<"TensorArrayCloseV3", []> {
      let summary = "Delete the TensorArray from its resource container.";
    
      let description = [{
    This enables the user to close and release the resource in the middle
    of a step/run.
      }];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top