- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 93 for getAxes (0.14 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/fuse_convolution_pass.cc
}); } // For dynamic case, the result of conv should be used by shape_of and mul. if (is_dynamic_broadcast) { auto conv_uses = (*conv_op.getODSResults(0).begin()).getUses(); if (std::distance(conv_uses.begin(), conv_uses.end()) != 2 || quant::FindUserOfType<shape::ShapeOfOp>(conv_op) == nullptr || quant::FindUserOfType<mhlo::MulOp>(conv_op) == nullptr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 22:21:19 UTC 2024 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc
for (const auto& p : llvm::zip(live_outs, launch_op.getResults())) { Value from = std::get<0>(p); // TODO(jingpu): move this to RegionUtils.h in MLIR core. for (auto& use : llvm::make_early_inc_range(from.getUses())) { if (launch_op_region->isAncestor(use.getOwner()->getParentRegion())) continue; use.set(std::get<1>(p)); } } } // Get all escaped live-out values of a region.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 05 13:30:21 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc
mlir::isa<IntegerType>(key_dtype) && mlir::cast<IntegerType>(key_dtype).getWidth() == 64))) { return false; } for (auto& use : hashtable->getUses()) { Operation* user = use.getOwner(); // Allow consuming hash table ops that can be covered by TensorFlow Lite // hash table kernels.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/group_by_dialect.cc
} // All results in our set that have a user outside our set. llvm::DenseSet<Value> outputs_seen; for (Operation* op : ops) { for (Value result : op->getResults()) { for (auto& use : result.getUses()) { if (!all_operations.contains(use.getOwner())) { if (!outputs_seen.contains(result)) { outputs->push_back(result); outputs_seen.insert(result); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 8K bytes - Viewed (0) -
pkg/kubelet/container/testing/fake_runtime.go
ContainerFilesystems: f.ContainerFsStats, } return resp, f.Err } func (f *FakeStreamingRuntime) GetExec(_ context.Context, id kubecontainer.ContainerID, cmd []string, stdin, stdout, stderr, tty bool) (*url.URL, error) { f.Lock() defer f.Unlock() f.CalledFunctions = append(f.CalledFunctions, "GetExec") return &url.URL{Host: FakeHost}, f.Err }
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed Feb 14 00:23:50 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
TF::TPUReplicatedInputOp private_input = input.clone(); builder.insert(private_input); forward_pass_ops.insert(private_input); for (OpOperand& next_use : input.getOutput().getUses()) { if (!forward_pass_ops.contains(next_use.getOwner())) continue; next_use.getOwner()->setOperand(next_use.getOperandNumber(), private_input.getOutput()); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc
// -------------------------------------------------------------------------- // namespace { constexpr char kDeviceAttr[] = "device"; // A type that abstracts over types that have uses accessible via `getUses`. using Source = PointerUnion<Operation *, BlockArgument *>; // We use union-find algorithm to build clusters of connected operations based // on the user provided policy. If an operation can be clustered (one of the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 27.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_dynamic_range.cc
ShapedType new_result_type = old_result_type.clone(quantized_type); // Insert CastOp if it does not exist yet. Otherwise, just rewire without // creating a CastOp. for (auto& connected_op : op.getResult().getUses()) { auto cast_op = llvm::dyn_cast_or_null<CastOp>(connected_op.getOwner()); if (cast_op && cast_op.getType() == new_result_type) { quantize_op->setOperand(quantize_operand_num, cast_op); return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.8K bytes - Viewed (0) -
platforms/core-execution/workers/src/main/java/org/gradle/workers/internal/WorkerDaemonClientsManager.java
synchronized (lock) { List<WorkerDaemonClient> sortedClients = CollectionUtils.sort(idleClients, comparingInt(WorkerDaemonClient::getUses)); List<WorkerDaemonClient> clientsToStop = selectionFunction.transform(new ArrayList<>(sortedClients)); if (!clientsToStop.isEmpty()) { stopWorkers(clientsToStop); }
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Thu May 30 19:54:37 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc
// determine whether we have taken care of all uses and can remove // the arg. for (BlockArgument arg : func.getArguments()) { auto uses = arg.getUses(); use_count[arg.getArgNumber()] = std::distance(uses.begin(), uses.end()); argument_to_index.insert({arg, arg.getArgNumber()}); } // We're only considering return values that are the same (at this
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0)