- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for resource_type (0.15 sec)
-
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/sparsecore/embedding_pipelining.cc
namespace mlir { namespace TFDevice { namespace { bool IsResourceType(Type val_type) { if (auto tensor_type = mlir::dyn_cast<mlir::TensorType>(val_type)) { if (mlir::isa<TF::ResourceType>(tensor_type.getElementType())) { return true; } } return false; } struct EmbeddingPipeliningPass : public ::impl::EmbeddingPipeliningPassBase<EmbeddingPipeliningPass> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md
This pass promotes resource accesses in function(s) (by default, the main) to input arguments and outputs of the function(s). Two types of resources are supported: (1) A function argument of TF::ResourceType type (this pass). (2) A VarHandleOp in the function (tf-promote-var-handles-to-args). After the pass, . The function will have an input argument for each resource that is
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0)