- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for setLoc (0.16 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
auto erf_op = rewriter.create<chlo::ErfOp>(mul_op.getLoc(), mul_op); auto add_op = rewriter.create<stablehlo::AddOp>(erf_op.getLoc(), erf_op, one_cst); auto lhs_mul_op = rewriter.create<stablehlo::MulOp>( half_cst.getLoc(), new_func.getArguments()[0], half_cst); auto output_mul_op = rewriter.create<stablehlo::MulOp>(lhs_mul_op.getLoc(),
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/tensorflow/transforms/initialize_variables_in_session_init.cc
builder.insert(var_handle_op_in_init); auto const_op = builder.create<mlir::arith::ConstantOp>( session_init_func.getLoc(), tensor_attr.getType(), tensor_attr); builder.create<TF::AssignVariableOp>( session_init_func.getLoc(), llvm::ArrayRef<mlir::Type>{}, llvm::ArrayRef<mlir::Value>{var_handle_op_in_init->getResult(0), const_op.getResult()}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/rewrite_tpu_embedding_ops.cc
SendTPUEmbeddingGradientsOp send_op; if (failed(GetOp(region, &send_op))) return failure(); // No TPU embedding ops. if (!recv_op && !send_op) return success(); Location loc = recv_op ? recv_op.getLoc() : send_op.getLoc(); StringRef config = recv_op ? recv_op.getConfig() : send_op.getConfig(); // Create XlaRecvTPUEmbeddingDeduplicationData op. OpBuilder builder(region); auto output_ty =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 22:55:42 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
return failure(); } auto op = builder.create<CustomOp>( func_.getLoc(), func_.getFunctionType().getResults(), func_.getArguments(), kMaxUnpooling, CustomOption(&builder, custom_option_buffer)); builder.create<func::ReturnOp>(func_.getLoc(), op.getResults()); return success(); } LogicalResult ConvertMaxUnpoolingFunc::VerifySignature() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/constant_fold.cc
return failure(); } results.clear(); builder.setInsertionPointAfter(op); for (const auto& result_value : result_values) { results.push_back(builder.create<TF::ConstOp>(op->getLoc(), result_value)); } return success(); } bool IsOperationFoldable(Operation* op) { if (isa<TF::ConstOp>(op)) return true; if (op->getDialect()->getNamespace() != "tf" || !TF::CanBeFolded(op)) {
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/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc
ArrayAttr ids_attr = builder.getStrArrayAttr(ids); ArrayAttr calibration_methods_attr = builder.getI32ArrayAttr(calibration_methods); builder.create<TF::CalibrationStatisticsSaverOp>( region.getLoc(), statistics_outputs, output_file_path_attr, ids_attr, calibration_methods_attr); return success(); } // Returns true if the op contains a `CalibrationStatisticsSaverOp`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
if (it.second) { auto resource_type = op.getResult(0).getType(); op.getResult(0).replaceAllUsesWith( block.addArgument(resource_type, op.getLoc())); AddEntryFunctionInput( absl::StrCat("hash_table_", it.first->getValue(), ":0"), target_func); // Avoid deleting the op here, clone it to the caller function first.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0)