- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 89 for getOperands (0.15 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_restore_op.cc
for (auto assign_variable_op : llvm::make_early_inc_range( session_init_func.getOps<TF::AssignVariableOp>())) { Value resource_operand = assign_variable_op.getOperand(0); Value assigned_value_operand = assign_variable_op.getOperand(1); if (auto var_handle_op = dyn_cast<TF::VarHandleOp>(resource_operand.getDefiningOp()); var_handle_op &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Mar 12 06:02:20 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
session_init_func.getOps<TF::AssignVariableOp>())) { Value resource_operand = assign_variable_op.getOperand(0); auto var_handle_op = dyn_cast<TF::VarHandleOp>(resource_operand.getDefiningOp()); if (!var_handle_op) continue; Value assigned_value_operand = assign_variable_op.getOperand(1); auto const_op = dyn_cast<TF::ConstOp>(assigned_value_operand.getDefiningOp());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/fold_constant_transpose.cc
public: using OpRewritePattern<mlir::stablehlo::TransposeOp>::OpRewritePattern; LogicalResult match(mlir::stablehlo::TransposeOp op) const override { Value operand = op.getOperand(); auto const_op = dyn_cast_or_null<mlir::stablehlo::ConstantOp>(operand.getDefiningOp()); if (!const_op) return failure(); // Only support float tensors.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/variables_utils.cc
ShapedType type; if (llvm::isa<TF::ReadVariableOp>(op)) { type = op->getResult(0).getType().cast<ShapedType>(); } else if (llvm::isa<TF::AssignVariableOp>(op)) { type = op->getOperand(1).getType().cast<ShapedType>(); } else if (llvm::isa<TF::VarHandleOp>(op)) { type = op->getResult(0) .getType() .cast<ShapedType>() .getElementType()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 21 19:32:03 UTC 2021 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc
output_shape[1] = composite_result_shape[2]; output_shape[2] = composite_result_shape[3]; output_shape[3] = composite_result_shape[1]; auto input_type = mlir::cast<ShapedType>(old_op->getOperand(0).getType()); return RankedTensorType::get(output_shape, input_type.getElementType()); } } // namespace odml
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 18:33:05 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_while_results.cc
// result all correspond to each other (see definition of `WhileRegionOp`). int idx = result.getResultNumber(); Value body_yield_operand = body_yield_op->getOperand(idx); Value body_block_argument = body_block.getArgument(idx); Value cond_block_argument = cond_block.getArgument(idx); // Consider the op that defines the unused result as a candidate for pruning.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 16 01:49:07 UTC 2022 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/hoist_broadcast_read.cc
} // `is_cpu_read` is set to `true` iff `read` is on a resource with device type // CPU. LogicalResult IsCpuRead(FuncOp func, ReadVariableOp read, bool& is_cpu_read) { if (auto arg = mlir::dyn_cast<BlockArgument>(read->getOperand(0))) { if (arg.getOwner() != &(func.front())) { is_cpu_read = false; return success(); } if (auto attr = func.getArgAttrOfType<StringAttr>(arg.getArgNumber(),
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/lite/transforms/split_merged_operands.cc
std::vector<int> stateful_operands_index; if (!IsStatefulOp(op, &stateful_operands_index)) return success(); for (int index : stateful_operands_index) { Value operand = op->getOperand(index); auto inserted_value = values->insert(operand).second; if (inserted_value) continue; // We can only clone the constant op or const->dequantize combo. The latter
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc
// parallel_execute region. assign_var.getOperation()->moveBefore(terminator); assign_var.getValueMutable().assign( terminator->getOperand(result.index())); results_to_remove.push_back(result.index()); } rewrite |= !results_to_remove.empty(); } if (!rewrite) return;
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/quantization/stablehlo/passes/post_quantize.cc
LogicalResult matchAndRewrite(mlir::stablehlo::UniformQuantizeOp op, PatternRewriter& rewriter) const override { DenseFPElementsAttr attr; if (matchPattern(op.getOperand(), m_Constant(&attr))) { const Type qtype = op.getResult().getType(); ElementsAttr quantized_attr = Quantize(attr, qtype); if (quantized_attr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 6.3K bytes - Viewed (0)