Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for VariableShape (0.27 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

        ```mlir
        tf_device.replicate([%0, %1] as %ri: tensor<*x!tf_type.resource>) {n = 2 : i32} {
          %2 = "tf.ReadVariableOp"(%ri) : tensor<*x!tf_type.resource> -> tensor<*xi32>
          %3 = "tf.VariableShape"(%0) : (tensor<*x!tf_type.resource>) -> tensor<?xi32>
          tf_device.return
        }
        ```
      }];
    
      let dependentDialects = [
        "mhlo::MhloDialect",
        "tf_device::TensorFlowDeviceDialect"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

    //===----------------------------------------------------------------------===//
    // ShapeOp
    //===----------------------------------------------------------------------===//
    
    namespace {
    // Validates Shape/ShapeN/VariableShape operand and associated result types.
    LogicalResult VerifyShapeOperandAndResult(Operation *op, Type operand_type,
                                              Type result_type,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
Back to top