- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 126 for func_20 (0.1 sec)
-
tensorflow/compiler/mlir/lite/utils/perception_ops_utils.h
public: explicit ConvertMaxUnpoolingFunc(func::FuncOp func, mlir::TF::FuncAttr attr) : func_(func), attr_(attr) {} LogicalResult RewriteFunc(); LogicalResult VerifySignature(); private: LogicalResult CreateCustomOptions(std::string& custom_option_buffer); func::FuncOp func_; mlir::TF::FuncAttr attr_; }; // Fuse DenseImageWarp ops annotated by tf.function to a TFLite custom op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 07 17:42:55 UTC 2022 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/verify_input_dialect_to_executor_pass.cc
return op->getName().getStringRef().str() == kClusterFuncOpName; } void VerifyInputDialectToExecutorPass::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { if (!tensorflow::tf2xla::internal::IsInBridgeAcceptableDialects(op)) { std::string error = "op is in dialect " +
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 08 16:32:56 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tfxla_device_specific_transforms.cc
get_alg_op->erase(); return success(); } void TFXLADeviceSpecificTransforms::runOnOperation() { if (!device_type_.hasValue()) return; auto func_op = getOperation(); auto walk_result = func_op->walk([&](TF::StatelessRandomGetAlgOp op) { if (failed(ConvertGetAlgOp(op))) { op->emitOpError( "Could not convert and remove Device specific information");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 05:56:39 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/verify_clustering_pass.cc
: public impl::VerifyClusteringPassBase<VerifyClusteringPass> { public: void runOnOperation() override; }; void VerifyClusteringPass::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { if (!tensorflow::tf2xla::internal::IsInBridgeAcceptableDialects(op)) { std::string error = "op is in dialect " +
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 20:33:37 UTC 2023 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/input_lowering_metrics_pass.cc
public: void runOnOperation() override; }; void InputMetricsLoweringPass::runOnOperation() { bool has_dynamic_op = false; Operation* func_op = getOperation(); func_op->walk([&](Operation* op) { auto abstractOp = op->getRegisteredInfo(); if (!abstractOp) return WalkResult::advance(); if (mlir::mhlo::IsDynamicPadderOp(abstractOp->getTypeID())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 08:55:35 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_import.cc
module_op->walk([&](func::FuncOp func_op) { FunctionName func_name = func_op.getSymName().str(); existing_func_names.insert(func_name); // We may retrieve the original function's name from the attribute. // Functions without this attribute are ignored. auto original_func_name = func_op->getAttrOfType<StringAttr>("tf._original_func_name"); if (original_func_name) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
it->second.data_type; } } } func_op.eraseArguments(indices_to_erase); func_op.setType( FunctionType::get(func_op.getContext(), new_types, llvm::to_vector<4>(return_op->getOperandTypes()))); } // Lifts reads/writes of resource arguments from func_op and changes its // signature. resource_data_types is the (index, data type) pair for each
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/transforms/executor_tpuv1_island_coarsening.cc
for (func::FuncOp func_op : getOperation().getOps<func::FuncOp>()) { if (functions_to_skip.contains(func_op)) { OpBuilder builder(func_op); // Mark this function as being skipped in island outlining. func_op->setAttr(mlir::TF::kSkipIslandOutlining, builder.getBoolAttr(true)); continue; } func_op.walk([&](GraphOp graph) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc
return {OperandIdx...}; } }; // Finds all op of type `T::OpType` `func_op` and recursively duplicates // constants used at the op's operands at `T::OperandIndices()`. It sequentially // does the same thing for `Ts`. template <typename T, typename... Ts> void DuplicateShapeDeterminingConstants(func::FuncOp func_op) { for (auto op : func_op.getOps<typename T::OpType>()) { RecursivelyDuplicateConstantsForOperands(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 17.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc
// Returns a list of "tf.entry_function" attribute's "inputs" comma-split // values. // // Example: if `func_op` has attribute `tf.entry_function = {inputs = // "arg0:0,arg1:0"}`, then this function returns `{"arg0:0", "arg1:0"}`. SmallVector<StringRef> GetEntryFunctionInputs(func::FuncOp func_op) { auto entry_function_attr = func_op->getAttrOfType<DictionaryAttr>("tf.entry_function"); SmallVector<StringRef> inputs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0)