- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for handle_dtype (0.42 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)