- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 263 for created (0.29 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/cc/quantization_unit_loc.cc
// If the op is rewritten, FusedLoc can be created. for (Location child_loc : mlir::cast<FusedLoc>(loc).getLocations()) { std::optional<QuantizationUnitLoc::QuantizationUnit> found_unit = FindQuantizationUnitFromLoc(child_loc); if (found_unit.has_value()) return found_unit; } } else if (isa<CallSiteLoc>(loc)) { // If the graph is inlined, CallSiteLoc can be created. return FindQuantizationUnitFromLoc(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_island_coarsening.cc
OpBuilder builder(merged_island.insert_point); auto new_island = builder.create<IslandOp>( merged_island.insert_point->getLoc(), result_types, operands); new_island.getBody().push_back(new Block); return new_island; } // Creates respective YieldOp for the new merged island. YieldOp CreateNewIslandYieldOp(IslandOp new_island,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
// creates a new session_initializer op and returns it. SessionInitializerOp GetOrCreateSessionInitializerOp(ModuleOp module_op) { SessionInitializerOp session_init_op = GetSessionInitializerOp(module_op); // Create one if it doesn't exist. if (!session_init_op) { OpBuilder builder(&module_op.getBodyRegion()); session_init_op = builder.create<SessionInitializerOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc
const Type expressed_type = value.getType(); // The value needs to be requantized. A Quantize op will be created to use // it as the operand and replace its uses. const Type new_type = state.params.castFromExpressedType(expressed_type); if (!new_type) return; auto requantize_op = builder_.create<quantfork::QuantizeCastOp>(loc, new_type, value); value.replaceAllUsesWith(requantize_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 38.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc
if (sub_op) { sub_op->replaceAllUsesWith(island.getOutputs()); sub_op->moveBefore(block, block->begin()); island_builder.create<tf_executor::YieldOp>(loc, sub_op->getResults()); } else { island_builder.create<TF::NoOp>(island.getLoc(), TypeRange{}, ValueRange{}); island_builder.create<tf_executor::YieldOp>(loc, ValueRange{}); } return island; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/instrumentations/save_report_test.cc
report_file_path)); const LogicalResult run_result = pm.run(*module_op); ASSERT_TRUE(succeeded(run_result)); // The report file is not created because `QuantizeCompositeFunctionsPass` was // not run. EXPECT_THAT(ReadFileToString(report_file_path), StatusIs(absl::StatusCode::kNotFound)); } TEST_F(SaveQuantizationReportInstrumentationTest,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/quantize_passes.cc
mlir::quant::stablehlo::FillPresetQuantizationOptions( quantization_options); } // TODO: b/276999414 - Add activation and bias quantization component as // respective quantization passes are created. QuantizationComponentSpec weight_component; for (const auto& component : quantization_options_.quantization_method() .custom_quantization_method()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 08:32:43 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/clustering_bridge_passes.cc
// Guarantee all functions have one use, which enables more exact shape // inference. pm.addPass(mlir::TF::CreateGuaranteeAllFuncsOneUsePass()); // Run shape inference so that tf_executor/tf_device ops created later will // likely to inherit more concrete types. pm.addPass(mlir::TF::CreateTFShapeInferencePass()); pm.addNestedPass<FuncOp>(mlir::TFTPU::CreateTPUPartitionedOpConversionPass()); pm.addNestedPass<FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 16:09:14 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/c/c_api.cc
"and will trigger an error in the future. Either don't modify " "nodes after running them or create a new session."); } } } namespace { // Helper method that creates a shape handle for a shape described by dims. tensorflow::shape_inference::ShapeHandle ShapeHandleFromDims( tensorflow::shape_inference::InferenceContext* ic, int num_dims,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
return tokens[0]; } else { return builder.create<AfterAllOp>(loc, original_token.getType(), tokens) .getResult(); } } // Replaces `tf._XlaHostComputeMlir` with individual `mhlo.send` and `mhlo.recv` // ops per operand and result. Unique Channel IDs are assigned per transfer. // Sink tokens are created across all `mhlo.send` ops first and then by // all `mhlo.recv` ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0)