- Sort Score
- Result 10 results
- Languages All
Results 131 - 140 of 234 for eraseOp (0.25 sec)
-
tensorflow/c/eager/tape.h
it->second--; if (it->second != 0) { return; } tensor_usage_.erase(it); auto tensor_op_it = tensor_tape_.find(tensor_id); if (tensor_op_it == tensor_tape_.end()) { return; } const int64_t op_id = tensor_op_it->second; if (op_id == -1) { // Do not delete watched tensors. return; } tensor_tape_.erase(tensor_op_it); auto op_it = op_tape_.find(op_id);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 12:40:29 UTC 2024 - 47.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
} args.append(new_args.begin(), new_args.end()); } else { args.append(yield_op->operand_begin(), yield_op->operand_end()); } b.create<func::ReturnOp>(yield_op->getLoc(), args); yield_op->erase(); SymbolTable(region.getParentOfType<ModuleOp>()).insert(outlined_func); outlined_func.setPrivate(); return outlined_func; } // Replace region with call to outline function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.h
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/get_alternative_subgraph.cc
// see if we need to erase the func op. // Ideally it would be nice if we can utilize dynamic illegal op to do // the job. if (device_inference_type.hardware != "CPU" && !IsAllSupportedbySpec(cloned_func, device_inference_type)) { cloned_func.erase(); } } } // Perform the device-specific optimization last.
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/tensorflow/transforms/functional_control_flow_to_cfg.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 13 11:42:59 UTC 2023 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
builder->setInsertionPoint(old_terminator); builder->create<mlir::tf_device::ReturnOp>(old_terminator->getLoc(), new_cluster_results); old_terminator->erase(); builder->setInsertionPoint(cluster); llvm::SmallVector<Type, 4> new_cluster_result_types; new_cluster_result_types.reserve(new_cluster_results.size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/raise_custom_ops.cc
} custom_op->setAttrs(inner_op->getAttrs()); builder.create<YieldOp>(loc, inner_op->getResults()); custom_op.getBody().takeBody(region); op->replaceAllUsesWith(custom_op); op->erase(); } } } // namespace std::unique_ptr<OperationPass<func::FuncOp>> CreateRaiseCustomOpsPass() { return std::make_unique<RaiseCustomOpsPass>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_duplicate_resource_ops.cc
op.replaceAllUsesWith(existing_resource->getParentOp()->getResults()); ops_to_remove.push_back(&op); } // Remove op after the loop to avoid crash. for (Operation* op : ops_to_remove) { op->erase(); } } static PassRegistration<MergeDuplicateResourceOpsPass> pass{}; } // namespace std::unique_ptr<OperationPass<func::FuncOp>> CreateMergeDuplicateResourceOpsPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 26 04:26:16 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
TransposeOp transpose = tranpose_op; transpose.getOperation()->moveBefore(op); transpose.setOperand(0, operand->get()); transpose.setOperand(1, permutation_op); transpose_ops->erase(it); return transpose; } } } return nullptr; } #define GEN_PASS_DEF_LAYOUTASSIGNMENTPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/weight_only_ptq.cc
*ctx, *module)); // Remove the `tpu` tag for exporting because the output quantized model is // essentially a CPU model. tags.erase("tpu"); py_function_library.SaveExportedModel( dst_saved_model_path, post_calibrated_exported_model, src_saved_model_path, tags, signature_def_map); return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 5.1K bytes - Viewed (0)