- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 93 for getAxes (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/replicate_tensor_list_init_ops_pass.cc
template <typename T> void ReplicateTensorListForUses(T tensor_list_op) { Value tensor_list = tensor_list_op.getResult(); std::vector<OpOperand*> uses; for (auto& use : tensor_list.getUses()) { uses.emplace_back(&use); } OpBuilder builder(tensor_list_op.getOperation()); for (OpOperand* operand : uses) { auto new_op = builder.clone(*tensor_list_op.getOperation());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Jan 22 17:28:34 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/constant_op_device_assignment.cc
} OpBuilder builder(op); llvm::StringMap<mlir::Operation *> cloned_op_by_device; bool all_uses_replaced = true; for (mlir::OpOperand &use : llvm::make_early_inc_range(op.getResult().getUses())) { mlir::Operation *user_op = use.getOwner(); StringAttr device_attr = user_op->getAttrOfType<StringAttr>(kDeviceAttr); if (!device_attr) { all_uses_replaced = false; continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 3.3K bytes - Viewed (0) -
pkg/volume/iscsi/iscsi.go
secret, err := createSecretMap(spec, plugin, pod.Namespace) if err != nil { return nil, err } return plugin.newMounterInternal(spec, pod.UID, &ISCSIUtil{}, plugin.host.GetMounter(plugin.GetPluginName()), plugin.host.GetExec(plugin.GetPluginName()), secret) }
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Tue May 14 06:17:25 UTC 2024 - 23.4K bytes - Viewed (0) -
pkg/controller/volume/persistentvolume/volume_host.go
return func(types.UID) { //nolint:logcheck klog.ErrorS(nil, "DeleteServiceAccountToken unsupported in PersistentVolumeController") } } func (adc *PersistentVolumeController) GetExec(pluginName string) utilexec.Interface { return utilexec.New() } func (ctrl *PersistentVolumeController) GetNodeLabels() (map[string]string, error) {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Thu Apr 18 11:00:37 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/topological_sort.cc
Value value = todo.front(); todo.pop(); // All operations that have all their inputs available are good to go. // Uses, not Users, in case getUsers ever dedups. for (OpOperand& operand : value.getUses()) { Operation* user = ancestor[operand.getOwner()]; remaining_incoming_data_edges[user]--; if (remaining_incoming_data_edges[user] == 0 && remaining_incoming_ctrl_edges[user] == 0) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 08 17:01:11 UTC 2022 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc
// before or above the parallel_execute. TF::AssignVariableOp GetSingleUseResourceWrite( tf_device::ParallelExecuteOp parallel_execute, Value result) { if (!result.hasOneUse()) return nullptr; OpOperand& use = *result.getUses().begin(); auto assign_var = dyn_cast<TF::AssignVariableOp>(use.getOwner()); if (!assign_var) return nullptr; if (use.get() != assign_var.getValue()) return nullptr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 06 04:46:18 UTC 2022 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/ConvertSimQuant.cc
for (auto m : fqOp.getMax()) max.push_back(cast<FloatAttr>(m).getValueAsDouble()); return fakeQuantAttrsToType(fqOp.getLoc(), fqOp.getNumBits(), fqOp.getAxis(), min, max, fqOp.getNarrowRange(), expressedType, fqOp.getIsSigned()); } }; void ConvertSimulatedQuantPass::runOnOperation() { bool hadFailure = false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/hoist_broadcast_read.cc
Value res = read.getResource(); Operation* scope = res.getParentBlock()->getParentOp(); if (!scope->isProperAncestor(replicate)) continue; bool has_conflicting_write = false; for (OpOperand& use : res.getUses()) { Operation* using_op = use.getOwner(); if (using_op == read) continue; if (!replicate->isProperAncestor(using_op)) continue; Operation* peer = GetAncestorBelow(using_op, replicate);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
// multiple such activations, one is returned (with no guarantee as to which // one). If there are no activation functions that use the output, returns // nullptr. Operation *GetActivation(Value op) { for (auto &use : op.getUses()) { if (IsActivationFunction(use.getOwner())) return use.getOwner(); } return nullptr; } // Finds and returns a BiasAdd that uses the result of `op` as the `value`
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/propagate_quantize_type.cc
llvm::SmallVector<Operation*> users(op->getUsers().begin(), op->getUsers().end()); bool changed = false; for (auto& use : op->getUses()) { Operation* user_op = use.getOwner(); int user_idx = use.getOperandNumber(); if (!IsOpWithInt8TypeOperand(user_op)) continue; // If the next op is terminator, function type needs to be changed so
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7K bytes - Viewed (0)