- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 80 for getAxes (0.18 sec)
-
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc
// Unsure if this check is required. if (state.pos != pos) { return; } } if (pos == RequantizeState::ON_OUTPUT) { Operation* user = value.getUses().begin().getUser(); if (isa<quantfork::QuantizeCastOp>(user)) { // The requantize op is inserted between `quantize` and `dequantize` ops. value = user->getResult(0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 38.1K bytes - Viewed (0) -
pkg/volume/util/resize_util.go
func GenericResizeFS(host volume.VolumeHost, pluginName, devicePath, deviceMountPath string) (bool, error) { resizer := mount.NewResizeFs(host.GetExec(pluginName)) return resizer.Resize(devicePath, deviceMountPath)
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Jul 17 19:30:35 UTC 2023 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// to it have the same shape, we need to follow use-def chain of ops that // (conceptually) modify it i.e., ops that take an input TensorList and // produce an output TensorList. for (auto& use : tensorlist.getUses()) { if (auto push = llvm::dyn_cast<TensorListPushBackOp>(use.getOwner())) { auto element_type = mlir::dyn_cast<RankedTensorType>(push.getTensor().getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
pkg/controller/volume/expand/expand_controller.go
} func (expc *expandController) GetMounter(pluginName string) mount.Interface { return nil } func (expc *expandController) GetExec(pluginName string) utilexec.Interface { return utilexec.New() } func (expc *expandController) GetHostName() string { return "" } func (expc *expandController) GetHostIP() (net.IP, error) {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Sat May 04 18:33:12 UTC 2024 - 16.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc
Value GetResultInBlock(Value value, llvm::DenseSet<Block*> target_blocks) { while (target_blocks.find(value.getParentBlock()) == target_blocks.end()) { Value new_value = nullptr; for (OpOperand& operand : value.getUses()) { Operation* owner = operand.getOwner(); if (!llvm::isa<tf_device::ReturnOp>(owner)) continue; Operation* parent = owner->getParentOp(); // op that owns the "return"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc
} return_operands.push_back(resource.live_value); result_types.push_back(resource.live_value.getType()); } if (is_var_handle && !resource.read) { assert(block.getArgument(index).getUses().empty()); argument_indices_to_remove.push_back(index); } else { if (is_var_handle) { // Add resource_name attribute to VarHandleOp read. function.setArgAttr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
pkg/volume/testing/volume_host.go
return func(namespace, name string) (*v1.Secret, error) { return f.kubeClient.CoreV1().Secrets(namespace).Get(context.TODO(), name, metav1.GetOptions{}) } } func (f *fakeVolumeHost) GetExec(pluginName string) exec.Interface { return f.exec } func (f *fakeVolumeHost) GetConfigMapFunc() func(namespace, name string) (*v1.ConfigMap, error) { return func(namespace, name string) (*v1.ConfigMap, error) {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Sat May 11 09:02:45 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc
// GraphOp, YieldOp and NextIterationSourceOp don't have control inputs so // exclude them below. for (Value out : island_op.getOutputs()) { for (auto& use : out.getUses()) { Operation* owner = use.getOwner(); if (owner->getDialect() == island_op->getDialect() && !llvm::isa<tf_executor::GraphOp, tf_executor::YieldOp,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
// OneHotOp //===----------------------------------------------------------------------===// LogicalResult OneHotOp::verify() { OneHotOp op = *this; int64_t axis = op.getAxis(); auto indices_ty = op.getIndices().getType().dyn_cast<RankedTensorType>(); if (indices_ty && !(axis == -1 || (axis >= 0 && axis <= indices_ty.getShape().size()))) { return op.emitOpError()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc
if (op.getNumOperands() != 0 || op.getNumResults() != 1 || !HasSingleOpInBlock<YieldOp>(&op.GetBody())) return failure(); for (auto &use : llvm::make_early_inc_range(op.getControl().getUses())) use.getOwner()->eraseOperand(use.getOperandNumber()); rewriter.eraseOp(op); return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 42.7K bytes - Viewed (0)