- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for TFTensorType (0.25 sec)
-
tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_mlrt_ops.td
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/gpu_ops.td
let description = [{ Transfer a CPU tensor to device. Example: %device_tensor = gpurt.transfer_to_device %cpu_tensor }]; let arguments = (ins TFTensorType); let results = (outs TFTensorType); let assemblyFormat = "operands attr-dict"; } // TODO(b/260267885): We may add a device argument when we want to support // GPU MIG. def TransferFromDeviceOp: Gpu_Op<"transfer_from_device"> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 18 15:01:21 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_sync.td
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 07 21:12:01 UTC 2023 - 7.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.cc
parse_options.has_cost = false; auto &builder = parser.getBuilder(); if (mlir::failed(fallback_common::ParseExecuteOpCommon( parser, builder, result, builder.getType<fallback::TFTensorType>(), parse_options))) return mlir::failure(); mlir::IntegerAttr num_args; if (parser.parseKeyword("num_args") || parser.parseLParen() ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 15.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.td
Note that this kernel always create a new AsyncValue for each result to avoid atomic contention on AsyncValue's refcount. }]; let arguments = (ins TFTensorType:$operand ); let results = (outs Variadic<TFTensorType>:$results ); let assemblyFormat = "$operand attr-dict `:` `(` type($operand) `)` `->` `(` type($results) `)`"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 18 15:01:21 UTC 2024 - 15.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback.td
}]; let cppNamespace = "::tfrt::fallback"; } // This corresponds to tensorflow::Tensor. def TFTensorType : DialectType<Fallback_Dialect, CPred<"$_self.isa<::tfrt::fallback::TFTensorType>()">, "!tfrt_fallback.tf_tensor type">, BuildableType<"$_builder.getType<::tfrt::fallback::TFTensorType>()"> { let description = [{ `!tfrt_fallback.tf_tensor type` represents a tensorflow::Tensor. }]; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 18 15:01:21 UTC 2024 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback.cc
if (keyword == "tf_tensor") return TFTensorType::get(getContext()); if (keyword == "tf_allocator") return TFAllocatorType::get(getContext()); parser.emitError(parser.getNameLoc(), "unknown type: ") << keyword; return Type(); } /// Print a type registered to this dialect. void FallbackDialect::printType(Type type, DialectAsmPrinter &os) const { if (mlir::isa<TFTensorType>(type)) { os << "tf_tensor"; return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback.h
Type parseType(DialectAsmParser &parser) const override; void printType(Type type, DialectAsmPrinter &os) const override; }; // The MLIR type represents a tensorflow::Tensor. class TFTensorType : public Type::TypeBase<TFTensorType, Type, TypeStorage> { public: using Base::Base; static constexpr StringLiteral name = "tfrt.tf_tensor"; }; // The MLIR type represents a tensorflow::Allocator. class TFAllocatorType
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 05 07:17:01 UTC 2023 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_ops.td
//===----------------------------------------------------------------------===// // Type definitions //===----------------------------------------------------------------------===// def TfTensorType : OpaqueType<"tfd", "tf_tensor", "!tfd.tf_tensor type">; //===----------------------------------------------------------------------===// // Runtime Fallback Dialect definitions
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 23 19:35:12 UTC 2023 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_mlrt_dialect.td
} // This corresponds to tensorflow::Tensor. def TFTensorType : DialectType<TensorflowMlrt_Dialect, CPred<"$_self.isa<::tensorflow::tf_mlrt::TFTensorType>()">, "!tf_mlrt.tensor type">, BuildableType<"$_builder.getType<::tensorflow::tf_mlrt::TFTensorType>()"> { let description = [{ `!tf_mlrt.tensor type` represents a tensorflow::Tensor. }]; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 18 15:01:21 UTC 2024 - 2K bytes - Viewed (0)