- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 127 for getDefiningOp (0.17 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc
cast<stablehlo::SubtractOp>(op.getOperand(0).getDefiningOp()); auto input1_i8_to_i32_convert_op = cast<stablehlo::ConvertOp>( input1_zero_point_subtract_op.getOperand(0).getDefiningOp()); auto input1_uniform_quantize_call_op = cast<func::CallOp>( input1_i8_to_i32_convert_op.getOperand().getDefiningOp()); auto input1_uniform_quantize_call_pattern =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/constant_fold.cc
SmallVector<ElementsAttr> inputs; for (auto operand : op->getOperands()) { auto preceding_const_op = operand.getDefiningOp<TF::ConstOp>(); if (preceding_const_op) { inputs.push_back(preceding_const_op.getValue()); continue; } Operation* preceding_op = operand.getDefiningOp(); int preceding_result_id = -1; for (auto preceding_result : preceding_op->getResults()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool_patterns.td
// See the function doc in the header file. def GetPadOpType : NativeCodeCall< "GetPadOpType((*$0.begin()).getDefiningOp<mhlo::CompositeOp>())">; // See the function doc in the header file. def GetAvgPoolOpPadAttr: NativeCodeCall<"GetAvgPoolOpPadAttr($_builder, (*$0.begin()).getDefiningOp<mhlo::CompositeOp>())">; // Returns true if the provided padding in the composite op can *not* be
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/defer_activation_transpose.cc
return IsTransposeOpWithPermuation(lhs.getDefiningOp(), kNhwcToNchwPermutation); } // Pushes the transpose op at the input to the result. void rewrite(mlir::stablehlo::ReduceWindowOp op, PatternRewriter& rewriter) const override { auto transpose_op = cast<TransposeOp>(op.getOperand(0).getDefiningOp());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc
queue.pop_back(); if (auto arg = mlir::dyn_cast_or_null<BlockArgument>(node)) { neighbors.push_back(arg.getArgNumber()); continue; } if (!node.getDefiningOp()) continue; for (Value operand : node.getDefiningOp()->getOperands()) { if (visited.contains(operand)) continue; queue.push_back(operand); visited.insert(operand); } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fold_broadcast_pass.cc
} template <typename Op, typename ElementType = Type, typename ValType, typename Convert> static Attribute BinaryFolder(Op *op) { auto lhs_op = op->getLhs().template getDefiningOp<mhlo::ConstantOp>(); auto rhs_op = op->getRhs().template getDefiningOp<mhlo::ConstantOp>(); if (!lhs_op || !lhs_op) return {}; auto lhs = dyn_cast_or_null<DenseElementsAttr>(lhs_op.getValue());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
dyn_cast_or_null<TFL::ExpOp>(sum_op.getInput().getDefiningOp()); if (!exp_op || !exp_op->hasOneUse()) { return failure(); } auto parent_sub_op = dyn_cast_or_null<TFL::SubOp>(sub_op.getLhs().getDefiningOp()); if (!parent_sub_op || parent_sub_op != dyn_cast_or_null<TFL::SubOp>( exp_op.getX().getDefiningOp())) { return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
if (block_arg_num.has_value()) return block_arg_num; Value next_input = input; auto pad_op = dyn_cast_or_null<TF::PadOp>(next_input.getDefiningOp()); auto cast_op = dyn_cast_or_null<TF::CastOp>(next_input.getDefiningOp()); while (pad_op || cast_op) { if (pad_op) { auto block_arg_num = GetBlockArgNum(pad_op.getInput()); if (block_arg_num.has_value()) return block_arg_num;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h
auto& tensor_property = enumerated_inputs.second; Value input = op.getOperand(index); if (input.getDefiningOp() == nullptr) continue; // TODO(b/172517537): make this work with non-PTQ case. if (llvm::isa<func::ConstantOp, arith::ConstantOp, TFL::ConstOp>( input.getDefiningOp())) { // Tensors with derived scale are biases, and handled in propagation.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 28K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
if (any_blargs) { return failure(); } auto tpose_arg1 = llvm::dyn_cast_or_null<TFL::TransposeOp>( op->getOperand(0).getDefiningOp()); auto tpose_arg2 = llvm::dyn_cast_or_null<TFL::TransposeOp>( op->getOperand(1).getDefiningOp()); if (!tpose_arg1 || !tpose_arg2) { return failure(); } auto tpose_arg1_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0)