- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for getOperandNumber (0.62 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
if (is_func && isa<func::ReturnOp>(user)) { assert(!info.IsResultIndexAssigned() && "Expect resource argument to returned no more than once"); info.result_index = use.getOperandNumber(); continue; } auto read = dyn_cast<TF::ReadVariableOp>(user); auto write = dyn_cast<TF::AssignVariableOp>(user); if (!read && !write) { unsupported_use = true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
for (OpOperand& operand : return_op->getOpOperands()) { if (non_tpu_args.contains(operand.getOperandNumber())) { if (BlockArgument argument = llvm::dyn_cast<BlockArgument>(operand.get())) { if (argument.getArgNumber() != operand.getOperandNumber()) { return_op.emitOpError() << "non TPU ops do not divide state into two pieces.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
auto return_op = llvm::dyn_cast<mlir::tf_device::ReturnOp>( device_cluster.GetBody().getTerminator()); for (OpOperand& operand : return_op->getOpOperands()) { auto operand_idx = operand.getOperandNumber(); return_operand_map[return_op.getOperand(operand_idx)].push_back( operand_idx); } // `return_value_from_host` and `return_value_from_device` together contain
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
// Tensorlist is passed into a while loop, check inside the body // function. auto inside_uses = while_op.body_function() .getArgument(use.getOperandNumber()) .getUses(); for (auto &inside_use : llvm::make_early_inc_range(inside_uses)) { if (TF::TensorListSetItemOp set_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
auto affine_user = llvm::dyn_cast<mlir::AffineQuantizedOpInterface>(user); if (affine_user && affine_user.GetAffineOperandIndex() == use.getOperandNumber() && affine_user.RequiredNarrowRangeAffineOperand()) continue; // Create a fully range quantized constant. if (full_range_const == value) { mlir::quant::QuantizedType new_qtype;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0)