- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 89 for getOperands (0.3 sec)
-
tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_combine.cc
mlir::tfd::ConvertTftToDhtOp op, mlir::PatternRewriter& rewriter) const override { // Look through the inputs of the ConvertTftToDhtOp. mlir::Value convert_op_input_0 = op.getOperand(0); mlir::Value convert_op_input_1 = op.getOperand(1); mlir::tfd::MoveDhtToTftOp move_input_op_0 = llvm::dyn_cast_or_null<mlir::tfd::MoveDhtToTftOp>( convert_op_input_0.getDefiningOp());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 09 12:09:19 UTC 2022 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
LogicalResult match(mlir::stablehlo::ConvolutionOp op) const override { // Handles 2D convolutions only. if (!HasRankOf(op.getOperand(0), /*rank=*/4) || !HasRankOf(op.getOperand(1), /*rank=*/4)) { return failure(); } if (!IsOpNotQuantized(op)) return failure(); const ConvDimensionNumbersAttr dimension_nums = op.getDimensionNumbers();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_reorder_replicate_and_partitioned_inputs.cc
Operation* pi = operand.getDefiningOp(); for (unsigned core_id = 0; core_id < num_cores_per_replica; ++core_id) { const auto pi_operand = packed_input ? pi->getOperand(0) : pi->getOperand(core_id); operands_per_replica_per_core[core_id].push_back(pi_operand); } } // Create new `tf.TPUReplicatedInput` ops feeding into one // `tf.TPUPartitionedInputV2` op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 24 23:08:55 UTC 2023 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/test_cluster_ops_by_policy.cc
if (is_op_a && *result_constraint == ValueConstraint::kShape) operands.Insert(op->getOperand(0), ValueConstraint::kRank); // `test.OpB` converts value constraint to shape constraint. if (*result_constraint == ValueConstraint::kValue) operands.Insert(op->getOperand(0), ValueConstraint::kShape); } return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 09:19:38 UTC 2022 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
if (!erf_user_user_input) return failure(); // Check `mul 0.5` and `mul 1/sqrt(2)` refer to the same input. if (erf_user_user_input->getOperand(0) != erf_input->getOperand(0)) { return failure(); } // Check the structural matches have the correct op type and values. auto rhs_mul = llvm::dyn_cast_or_null<stablehlo::MulOp>(erf_input);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/save_variables.cc
// variable is not saved, returns an empty string. absl::StatusOr<std::string> AddTensorToBundleWriter( mlir::TF::AssignVariableOp assign_var_op, BundleWriter& bundle_writer) { auto resource_operand = assign_var_op.getOperand(0); auto var_handle_op = llvm::dyn_cast<mlir::TF::VarHandleOp>(resource_operand.getDefiningOp()); if (!var_handle_op) { assign_var_op->emitRemark(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:36:55 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc
Value pad_input = pad_op.getOperand(); RankedTensorType pad_type = pad_op.getType().cast<RankedTensorType>(); auto transpose_op = pad_input.getDefiningOp<stablehlo::TransposeOp>(); if (!transpose_op || !transpose_op->hasOneUse()) return failure(); Value transpose_input = transpose_op.getOperand(); ArrayRef<int64_t> transpose_perm = transpose_op.getPermutation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/remove_var_init_by_const.cc
using OpRewritePattern<TF::AssignVariableOp>::OpRewritePattern; LogicalResult match(TF::AssignVariableOp assign_op) const override { Value resource_operand = assign_op.getOperand(0); Value assigned_value_operand = assign_op.getOperand(1); if (isa<TF::VarHandleOp>(resource_operand.getDefiningOp()) && isa<TF::ConstOp>(assigned_value_operand.getDefiningOp())) { return success(); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 03 12:04:03 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver_test.cc
Operation* xla_call_module_op = FindOperationOfType<TF::XlaCallModuleOp>(main_fn); Operation* filter_dcast_op = xla_call_module_op->getOperand(1).getDefiningOp(); Operation* filter_qcast_op = filter_dcast_op->getOperand(0).getDefiningOp(); ASSERT_NE(filter_qcast_op, nullptr); EXPECT_TRUE(isa<quantfork::QuantizeCastOp>(filter_qcast_op));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/propagate_quantize_type.cc
Operation* user_op, int user_idx, Type new_user_op_type) const { auto op_before_dequantize = original_dequantize_op.getOperand(0); // Create a new dequantize op that is propagated. rewriter.setInsertionPointAfter(user_op); TF::PartitionedCallOp new_dequantize_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7K bytes - Viewed (0)