- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 46 for resource_type (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc
for (BlockArgument& func_arg : func_args) { auto resource_type = mlir::dyn_cast<TF::ResourceType>( getElementTypeOrSelf(func_arg.getType())); if (!resource_type) continue; if (failed(ValidateResourceArgument(function, func_arg, resource_type))) return failure(); add_resource_argument(func_arg, resource_type); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
auto it = shared_name_to_arg_idx.insert({shared_name, block.getNumArguments()}); if (it.second) { auto resource_type = op.getResult(0).getType(); op.getResult(0).replaceAllUsesWith( block.addArgument(resource_type, op.getLoc())); AddEntryFunctionInput( absl::StrCat("hash_table_", it.first->getValue(), ":0"), target_func);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
auto resource_type = mlir::TF::ResourceType::get({resource_subtype}, builder.getContext()); auto tensor_type = mlir::cast<mlir::TensorType>(mlir_arg.getType()); if (tensor_type.hasRank()) { mlir_arg.setType( GetTypeFromTFTensorShape(tensor_type.getShape(), resource_type)); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
node_def->set_op(FunctionLibraryDefinition::kArgOp); mlir::TensorType arg_type = mlir::cast<mlir::TensorType>(arg.getType()); if (auto resource_type = mlir::dyn_cast<mlir::TF::ResourceType>(arg_type.getElementType())) { llvm::ArrayRef<mlir::TensorType> subtypes = resource_type.getSubtypes(); if (!subtypes.empty()) { AttrValue handle_dtypes_attr; AttrValue handle_shapes_attr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
node_def->set_op(FunctionLibraryDefinition::kArgOp); mlir::TensorType arg_type = mlir::cast<mlir::TensorType>(arg.getType()); if (auto resource_type = mlir::dyn_cast<mlir::TF::ResourceType>(arg_type.getElementType())) { llvm::ArrayRef<mlir::TensorType> subtypes = resource_type.getSubtypes(); if (!subtypes.empty()) { AttrValue handle_dtypes_attr; AttrValue handle_shapes_attr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
return op.emitOpError("requires input to have one resource"); auto resource_type = input_type.getElementType().cast<TF::ResourceType>(); auto subtypes = resource_type.getSubtypes(); switch (subtypes.size()) { case 1: return VerifyShapeOperandAndResult( op, resource_type.getSubtypes().front(), op.getType()); case 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) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
bool IsResource(Value value) { return mlir::isa<TF::ResourceType>(getElementTypeOrSelf(value.getType())); } // Get the type of the data contained in a resource. Returns null if there is // no single type in the resource. Type GetResourceSubtype(Value value) { auto resource_type = mlir::dyn_cast<TF::ResourceType>(getElementTypeOrSelf(value.getType())); auto subtypes = resource_type.getSubtypes();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
[{ mlir::TF::ShapeAttr::get($_ctxt, $_self) }]>; let extraClassDeclaration = [{ TensorType resource_subtype() { return resource_type().getSubtypes()[0]; } ResourceType resource_type() { return getElementTypeOrSelf(getResource()).cast<TF::ResourceType>(); } }]; let hasVerifier = 1; } def TF_XlaShardingOp : TF_Op<"XlaSharding", [Pure, TF_NoConstantFold]> {
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/mlir/tensorflow/transforms/shape_inference.cc
} TensorType resource_subtype = mlir::cast<TensorType>(value.getType()); ResourceType resource_type = ResourceType::get({resource_subtype}, op.getContext()); UnrankedTensorType new_resource_type = UnrankedTensorType::get(resource_type); Type refined_type = TypeMeet(resource.getType(), new_resource_type); if (refined_type == resource.getType()) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
pilot/pkg/xds/debug.go
Registered: Fri Jun 14 15:00:06 UTC 2024 - Last Modified: Tue Apr 30 00:26:45 UTC 2024 - 39.5K bytes - Viewed (0)