- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for CallableOpInterface (0.9 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_inline_tpu_island.cc
auto called_func = dyn_cast_or_null<func::FuncOp>(call_interface.resolveCallable()); if (failed(inlineCall(inliner, call_interface, cast<CallableOpInterface>(called_func.getOperation()), called_func.getCallableRegion(), /* shouldCloneInlinedRegion = */ false))) { call_op.emitOpError() << "Failed to inline\n";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/target_annotation.cc
break; } } } // default to CPU if (!device_is_set) { if (IsNonConstOp(op) && !IsTerminatorOp(op) && !llvm::isa<func::ReturnOp, func::FuncOp, CallableOpInterface>(op)) { SetAnnotation(op, kDevice, "CPU", builder); device_is_set = true; } } if (!device_is_set) { op->emitError("cannot set target device for this ops"); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 19:32:06 UTC 2023 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc
} } } }); // Find all functions module->walk([&](SymbolOpInterface op) { if (!op.isPrivate()) return; auto call = llvm::dyn_cast<CallableOpInterface>(op.getOperation()); if (!call) return; Region* region = call.getCallableRegion(); if (!region) return; // happens e.g. for external functions
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/get_alternative_subgraph.cc
// Set device for each op. cloned_func.walk([&](Operation* op) { if (IsNonConstOp(op) && !IsTerminatorOp(op) && !llvm::isa<func::ReturnOp, func::FuncOp, CallableOpInterface>(op)) { op->setAttr(kDevice, builder->getStringAttr( target_device_inference_type.hardware)); op->setAttr(kInferenceType,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
// composition. if (failed(inlineCall(inliner, cast<CallOpInterface>(call_op.getOperation()), cast<CallableOpInterface>(callee.getOperation()), callee.getCallableRegion(), /**shouldCloneInLinedRegion=*/true))) { // This failure is usually because the decompose function is not defined.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
auto operands = adaptor.getOperands(); (void)operands; assert(operands.empty() && "constant has no operands"); // Return the held attribute value. return getValue(); } // CallableOpInterface Region *TFRFuncOp::getCallableRegion() { return isExternal() ? nullptr : &getBody().front(); } //===----------------------------------------------------------------------===// // Dialect type definitions
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0)