- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 382 for get_operations (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference_pass.cc
input_shapes_ = input_shapes_vec; } auto failure_or_converged = InferModuleShape( getOperation(), max_iterations_, /*ops_to_skip=*/{}, input_shapes_); if (failed(failure_or_converged)) return signalPassFailure(); if (!failure_or_converged.value()) { getOperation().emitError() << "shape inference pass did not reach convergence after " << max_iterations_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/move_tpu_compile_to_front.cc
if (op->hasAttr("_wraps_compilation")) return 1; // Followed by everything else. return 0; } void MoveTpuCompileToFrontPass::runOnOperation() { MarkCompilationOps(getOperation()); getOperation().walk([](Operation* op) { for (Region& region : op->getRegions()) { for (Block& block : region.getBlocks()) { if (block.empty()) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 17 00:26:18 UTC 2023 - 3.2K bytes - Viewed (0) -
subprojects/core/src/testFixtures/groovy/org/gradle/internal/operations/TestBuildOperationExecutor.java
this.runner = buildOperationRunner; this.log = runner.log; } public List<BuildOperationDescriptor> getOperations() { return runner.getOperations(); } @Override public <O extends RunnableBuildOperation> void runAll(Action<BuildOperationQueue<O>> schedulingAction) {
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Wed Apr 03 12:12:49 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/name_anonymous_iterators.cc
return count; } void NameAnonymousIteratorsPass::runOnOperation() { int count = 1; getOperation().walk( [&](TF::AnonymousIteratorOp op) { count = replace(op, count); }); getOperation().walk( [&](TF::AnonymousIteratorV2Op op) { count = replace(op, count); }); getOperation().walk( [&](TF::AnonymousIteratorV3Op op) { count = replace(op, count); }); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/trim_functions_tf.cc
// SymbolRefAttrs of all ops are present. SymbolTable symbol_table = SymbolTable(getOperation()); llvm::SetVector<func::FuncOp> reachable_funcs; for (auto func : getOperation().getOps<func::FuncOp>()) { auto walk_result = func.walk([&](func::CallOp op) -> WalkResult { if (!symbol_table.lookup<func::FuncOp>(op.getCallee())) return getOperation().emitError() << func.getName() << " is not in the funcs allowlist";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 07 21:08:41 UTC 2022 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_assignment.cc
default_device_ = std::string(default_device); } void runOnOperation() override { Builder builder(&getContext()); Dialect* tf = getContext().getLoadedDialect<TensorFlowDialect>(); getOperation().walk([&](Operation* op) { if (auto device_attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { // We assign default device to ops with device attribute that is empty. if (device_attr.getValue().empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replica_id_to_device_ordinal.cc
const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); if (!tf_dialect) { getOperation().emitError() << "'tf' dialect is not registered"; return signalPassFailure(); } // Get the number of devices per host. int device_num = 0; mlir::TF::RuntimeDevices devices; if (failed(tensorflow::GetDevicesFromOp( getOperation()->getParentOfType<ModuleOp>(), &devices))) return signalPassFailure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.h
// Update types for all layout sensitive results. auto layout_sensitive = cast<LayoutSensitiveInterface>(op->getOperation()); for (unsigned idx : layout_sensitive.GetLayoutDependentResults()) { OpResult result = op->getOperation()->getResult(idx); result.setType(ShuffleRankedTensorType(result.getType(), perm)); } return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 5.3K bytes - Viewed (0) -
plugin/pkg/admission/noderestriction/admission.go
} return nil default: return admission.NewForbidden(a, fmt.Errorf("unexpected operation %s", a.GetOperation())) } } func (p *Plugin) admitPVCStatus(nodeName string, a admission.Attributes) error { switch a.GetOperation() { case admission.Update: oldPVC, ok := a.GetOldObject().(*api.PersistentVolumeClaim) if !ok {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Thu Mar 07 21:22:55 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/device_index_selector.cc
: public impl::DeviceIndexSelectorPassBase<DeviceIndexSelector> { void runOnOperation() override; }; } // namespace void DeviceIndexSelector::runOnOperation() { func::FuncOp func = getOperation(); // Convert all the DeviceIndex ops to constant values. func.getBody().walk([](TF::DeviceIndexOp op) { // This just selects the default in all cases where DeviceIndex feeds into
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 3.4K bytes - Viewed (0)