Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 24 for handle_dtype (0.36 sec)

  1. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

    }
    
    Type GetTensorTypeForTensorList(Type element_type, TF::VariantType handle_dtype,
                                    PatternRewriter *rewriter) {
      // If the variant type in the output handle has item shape available, use it
      // to derive the output shape by setting unknown leading dimension.
      // Otherwise, result type will be of unranked type.
      if (handle_dtype.getSubtypes().empty()) {
        return UnrankedTensorType::get(element_type);
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

      let extraClassDeclaration = [{
        // Returns type of the TensorList element produced by this op.
        TensorType element_type() { return handle_dtype().getSubtypes()[0]; }
    
        // Returns data type of the result handle. Returned type contains type of
        // the TensorList element as a subtype.
        VariantType handle_dtype() {
          return getElementTypeOrSelf(getHandle().getType()).cast<TF::VariantType>();
        }
      }];
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/shape_inference.cc

      if (a.handle_type == DT_INVALID) {
        result.handle_type = b.handle_type;
      } else if (b.handle_type == DT_INVALID) {
        result.handle_type = a.handle_type;
      } else if (a.handle_type == b.handle_type) {
        result.handle_type = a.handle_type;
      } else {
        return errors::InvalidArgument(
            "Mismatched resource types: ", DataTypeString(a.handle_type), " vs. ",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 00:41:19 UTC 2024
    - 13K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/function-resource-args-handle-info.mlir

         tf_executor.fetch %outputs : tensor<8x1xf32>
      }
      func.return %0 : tensor<8x1xf32>
    }
    
    // Check that we generate _handle_dtypes and _handle_shapes for the resource
    // argument.
    
    // CHECK:      op: "_Arg"
    
    // CHECK:        key: "_handle_dtypes"
    // CHECK-NEXT:   value {
    // CHECK-NEXT:     list {
    // CHECK-NEXT:       type: DT_FLOAT
    // CHECK-NEXT:     }
    // CHECK-NEXT:   }
    // CHECK-NEXT: }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 25 12:28:56 UTC 2022
    - 1.2K bytes
    - Viewed (0)
  5. subprojects/core/src/main/java/org/gradle/deployment/internal/DeploymentRegistry.java

         * @param handleType type of deployment handle
         * @param params constructor arguments
         *
         * @throws IllegalStateException if deployment handle with the given name already exists
         */
        <T extends DeploymentHandle> T start(String name, ChangeBehavior changeBehavior, Class<T> handleType, Object... params);
    
        /**
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Sat Mar 23 06:23:31 UTC 2019
    - 2.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

      // This is required to populate derived attributes during export in a
      // meaningful way. Else during export to GraphDef element_type() query
      // will result in out of bounds access/assert.
      if (handle_dtype().getSubtypes().size() != 1) {
        return emitOpError(
            "must have exactly one subtype in the result variant type");
      }
    
      if (!IsOfRankOrUnranked(op.getElementShape(), 0) &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

      // This is required to populate derived attributes during export in a
      // meaningful way. Else during export to GraphDef element_type() query
      // will result in out of bounds access/assert.
      if (handle_dtype().getSubtypes().size() != 1) {
        return emitOpError(
            "must have exactly one subtype in the result variant type");
      }
      if (!IsOfRankOrUnranked(op.getElementShape(), 0) &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  8. subprojects/core/src/main/java/org/gradle/deployment/internal/DefaultDeploymentRegistry.java

                lock.unlock();
            }
        }
    
        @Override
        public <T extends DeploymentHandle> T get(String name, Class<T> handleType) {
            lock.lock();
            try {
                failIfStopped();
                if (deployments.containsKey(name)) {
                    return Cast.cast(handleType, deployments.get(name).getHandle());
                } else {
                    return null;
                }
            } finally {
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Fri Mar 08 13:46:07 UTC 2024
    - 6.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc

        if (auto tensor_array_v3_op = dyn_cast<TF::TensorArrayV3Op>(tf_op)) {
          Value handle = tensor_array_v3_op.getHandle();
          auto handle_type = mlir::cast<TensorType>(handle.getType());
          if (handle_type.getElementType().isInteger(/*width=*/32)) {
            Type resource_tensor_type =
                handle_type.clone(TF::ResourceType::get(rewriter.getContext()));
            handle.setType(resource_tensor_type);
          }
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/tests/keras_imagenet_main.pbtxt

      op: "_Arg"
      device: "/job:localhost/replica:0/task:0/device:GPU:0"
      attr {
        key: "T"
        value {
          type: DT_RESOURCE
        }
      }
      attr {
        key: "_handle_dtypes"
        value {
          list {
            type: DT_FLOAT
          }
        }
      }
      attr {
        key: "_handle_shapes"
        value {
          list {
            shape {
              dim {
                size: 1001
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 30 02:52:54 UTC 2019
    - 1.3M bytes
    - Viewed (0)
Back to top