- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 142 for GetOperands (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
return RefineTypeForPassThroughOperands( op, iter_sink.getOperands().drop_front().take_front(), iter_source.getResults()); } if (auto launch_op = dyn_cast<tf_device::LaunchOp>(op)) { auto terminator = launch_op.GetBody().getTerminator(); return RefineTypeForPassThroughOperands(op, terminator->getOperands(), op->getResults()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
const OpSetVector& cluster_dependent_ops) { Operation* other_op = nullptr; op->walk([&](Operation* inner_op) { ValueRange values = incoming ? ValueRange(inner_op->getOperands()) : ValueRange(inner_op->getResults()); llvm::SmallVector<Operation*, 4> candidates; for (Value value : values) { if (incoming) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc
// so this op can be quantized. This is only applied on the returned result // because the error will not be accumulated. func.walk([&](func::ReturnOp ret) { int i = 0; for (Value returned : ret.getOperands()) { llvm::SmallVector<Value, 4> quantized; for (auto user : returned.getUsers()) { if (auto q = Quantized(user)) { quantized.push_back(q); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc
// so this op can be quantized. This is only applied on the returned result // because the error will not be accumulated. func.walk([&](func::ReturnOp ret) { int i = 0; for (Value returned : ret.getOperands()) { llvm::SmallVector<Value, 4> quantized; for (auto user : returned.getUsers()) { if (auto q = Quantized(user)) { quantized.push_back(q); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_validate_inputs.cc
} // Gets the predecessors of an op wrapped in tf_executor.island. llvm::SmallVector<Operation*> GetPredecessors(Operation* op) { llvm::SmallVector<Operation*> predecessors; for (auto operand : op->getOperands()) { if (Operation* pred = operand.getDefiningOp()) { pred->walk([&](mlir::Operation* opinexecutor) { predecessors.push_back(opinexecutor); }); } } return predecessors; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 06:51:01 UTC 2024 - 21.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
// inputs. SmallVector<Value, 4> inputs; inputs.reserve(quantizing_op->getNumOperands()); for (auto operand : quantizing_op->getOperands()) { Type operand_type = operand.getType(); if (operand_type.isa<NoneType>()) { inputs.push_back(operand); continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc
std::vector<tensorflow::Tensor>& tensors, std::vector<tensorflow::TensorValue>& inputs) { // Prepare the list of Tensor inputs for the kernel. for (auto it : llvm::enumerate(op_->getOperands())) { Value operand = it.value(); size_t idx = it.index(); tensorflow::XlaExpression expr = GetExprForOperand(operand, op_, idx); tensorflow::XlaExpression::Kind kind = expr.kind();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 18.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
OpBuilder builder(while_op); llvm::SmallVector<Type, 4> new_operand_types; llvm::SmallVector<Value> new_operands; auto operands = while_op->getOperands(); const int num_operands = while_op->getNumOperands(); llvm::BitVector skip_indices(num_operands); for (int i : arguments_to_erase) skip_indices.set(i); for (int i = 0; i < num_operands; ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
if (compare_op.getComparisonDirection() != mhlo::ComparisonDirection::GT) { return false; } if (compare_op.getOperands()[0] != comparator_blk.getArgument(0) || compare_op.getOperands()[1] != comparator_blk.getArgument(1)) { return false; } return return_op.getOperands().front() == compare_op.getResult(); } // In general, we convert the following form of sort to tf.TopK: //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc
value_queue.pop(); Operation* defining_node = current_value.getDefiningOp(); if (defining_node == nullptr) continue; op_stack.push(defining_node); for (Value arg : defining_node->getOperands()) { if (!argument_set.contains(arg.getImpl())) { value_queue.push(arg); } } } // Remove duplicate ops from the op stack. SmallVector<Operation*> sorted_ops;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0)