Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 15 for handle_dtype (0.42 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/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)
  5. 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)
  6. 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)
  7. 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)
  8. tensorflow/compiler/jit/shape_inference.h

    namespace tensorflow {
    
    struct InferredShape {
      // Shape of the argument tensor.
      PartialTensorShape shape;
    
      // If the argument is a resource variable, the type and shape of the
      // variable's value.
      DataType handle_type = DT_INVALID;
      PartialTensorShape handle_shape;
    };
    typedef std::unordered_map<string, std::vector<InferredShape>> GraphShapeInfo;
    
    // Infer shapes for all Tensors in a graph, and save them in a map.  The vector
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 00:41:19 UTC 2024
    - 2.2K bytes
    - Viewed (0)
  9. src/internal/syscall/windows/zsyscall_windows.go

    	ret = uintptr(r0)
    	return
    }
    
    func RtlVirtualUnwind(handlerType uint32, baseAddress uintptr, pc uintptr, entry uintptr, ctxt uintptr, data *uintptr, frame *uintptr, ctxptrs *byte) (ret uintptr) {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Thu Mar 21 11:49:46 UTC 2024
    - 18.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/shape_inference_test.cc

                               &shape_info));
      auto iter = shape_info.find("sink");
      EXPECT_NE(iter, shape_info.end());
      EXPECT_EQ(iter->second.size(), 1);
      EXPECT_EQ(iter->second.at(0).handle_type, DT_FLOAT);
      TensorShape resource_shape;
      EXPECT_TRUE(iter->second.at(0).handle_shape.AsTensorShape(&resource_shape));
      EXPECT_EQ(resource_shape, TensorShape({2, 3}));
    }
    
    }  // namespace
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 00:41:19 UTC 2024
    - 10.3K bytes
    - Viewed (0)
Back to top