- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 32 for addTypes (0.18 sec)
-
tensorflow/compiler/mlir/tfrt/ir/mlrt/mlrt_dialect.cc
} }; } // namespace MlrtDialect::MlrtDialect(mlir::MLIRContext *context) : mlir::Dialect(/*name=*/"mlrt", context, mlir::TypeID::get<MlrtDialect>()) { addTypes<FutureType>(); addTypes<PromiseType>(); addTypes<AsyncHandleType>(); addInterfaces<MlrtInlinerInterface>(); addOperations< #define GET_OP_LIST #include "tensorflow/compiler/mlir/tfrt/ir/mlrt/mlrt_ops.cpp.inc" >(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback.cc
namespace tfrt { namespace fallback { FallbackDialect::FallbackDialect(MLIRContext *context) : Dialect(/*name=*/"tfrt_fallback", context, TypeID::get<FallbackDialect>()) { addTypes<TFTensorType, TFAllocatorType>(); addOperations< #define GET_OP_LIST #include "tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback.cpp.inc" >(); } /// Parse a type registered to this dialect.
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/mlrt/tf_mlrt_ops.cc
}; } // namespace TensorflowMlrtDialect::TensorflowMlrtDialect(mlir::MLIRContext *context) : mlir::Dialect(/*name=*/"tf_mlrt", context, mlir::TypeID::get<TensorflowMlrtDialect>()) { addTypes<TFTensorType, TFDeviceType>(); addInterfaces<TensorflowMlrtInlinerInterface>(); addOperations< #define GET_OP_LIST #include "tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_mlrt_ops.cpp.inc" >();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc
#define GET_OP_LIST #include "tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc.inc" >(); addInterfaces<TensorFlowExecutorInlinerInterface, TensorFlowExecutorDialectFoldInterface>(); addTypes<ControlType, TokenType>(); } Type TensorFlowExecutorDialect::parseType(DialectAsmParser &parser) const { StringRef data_type; if (parser.parseKeyword(&data_type)) return Type();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 42.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
// Create the TF op const std::string tf_op_full_name = absl::StrCat("tf.", node_def.op()); mlir::OperationState op_state(loc, tf_op_full_name); op_state.addOperands(func.getArguments()); op_state.addTypes(output_tys); op_state.addAttributes(attrs); mlir::Operation* tf_op = op_builder.create(op_state); op_builder.create<mlir::func::ReturnOp>(loc, tf_op->getResults()); // Run the decompose passes on the module
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc
llvm::SmallVector<mlir::Type, 4> output_tys; for (int i = 0; i < custom_op.getNumResults(); i++) { output_tys.push_back(custom_op.getType(i)); } op_state.addTypes(output_tys); op_state.addAttributes(attr); auto stablehlo_op = builder.create(op_state); custom_op.replaceAllUsesWith(stablehlo_op); custom_op.erase(); }); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
const std::string tf_op_full_name = llvm::Twine("tf.", tf_op_name).str(); // Create the TF op OperationState op_state(op.getLoc(), tf_op_full_name); op_state.addOperands(op.getOperands()); op_state.addTypes(op.getResultTypes()); SmallVector<NamedAttribute, 2> attrs; std::string parsed_op_name; tensorflow::NodeDef node_def; if (failed(ParseCustomOption(op.getCustomOption().getValue(), op.getLoc(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
platforms/core-runtime/internal-instrumentation-processor/src/main/java/org/gradle/internal/instrumentation/processor/codegen/groovy/InterceptGroovyCallsGenerator.java
) { List<TypeSpec> interceptorTypeSpecs = generateInterceptorClasses(requestsClassGroup, onFailure); return builder -> builder .addModifiers(Modifier.PUBLIC) .addTypes(interceptorTypeSpecs); } @SuppressWarnings("ReturnValueIgnored")
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Tue May 28 13:09:40 UTC 2024 - 16.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
DCHECK_GE(num_regions, 1); for (int i = 0; i < num_regions; ++i) { Region* region = state.addRegion(); region->push_back(new Block); } state.addTypes(output_types); } Block& ParallelExecuteOp::GetRegionBlockWithIndex(unsigned index) { return getOperation()->getRegion(index).front(); } Operation::result_range ParallelExecuteOp::GetRegionOutputs(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td
bool WrapsSingleOp(); }]; let builders = [ OpBuilder<(ins "StringAttr":$device, "TypeRange":$result_types), [{ $_state.addAttribute("device", device); $_state.addTypes(result_types); $_state.addRegion(); }]> ]; let hasCanonicalizer = 1; } def TfDevice_ReturnOp : TfDevice_Op<"return", [Pure, ReturnLike, Terminator]> { let summary = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 14.8K bytes - Viewed (0)