- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 101 for getFlow (0.6 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
for (const auto& arg : func_op.getArguments()) { arg_locs.push_back(arg.getLoc()); } } // Creates a new main function. auto func_type = FunctionType::get(context, arg_types, result_types); auto main_func = builder.create<func::FuncOp>( module_op.getLoc(), kImportModelDefaultGraphFuncName, func_type); builder.createBlock(&main_func.getBody(), main_func.begin(), arg_types,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_island_coarsening.cc
} } auto island = OpBuilder(fetch).create<IslandOp>( fetch.getLoc(), data_types, /*control=*/ControlType::get(fetch.getContext()), /*controlInputs=*/control_fetches); island.getBody().push_back(new Block); OpBuilder::atBlockEnd(&island.GetBody()) .create<YieldOp>(fetch.getLoc(), data_fetches); const int fetch_control_idx = data_fetches.size();
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/tf2xla/internal/passes/tpu_cluster_formation.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
auto merged_execute_launch = builder->create<tf_device::LaunchOp>( merged_execute.getLoc(), execute_launch.getDeviceAttr(), merged_execute.getResultTypes()); merged_execute_launch.getBody().push_back(new Block); builder->setInsertionPointToEnd(&merged_execute_launch.GetBody()); builder->create<tf_device::ReturnOp>(merged_execute.getLoc(), merged_execute.getResults());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 17:52:11 UTC 2024 - 27K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc
return failure(); } auto tfl_fusable_op = builder.create<TFL::CustomOp>( func->getLoc(), func.getFunctionType().getResults(), func.getArguments(), custom_op_name, CustomOption(&builder, custom_option_buffer)); builder.create<func::ReturnOp>(func->getLoc(), tfl_fusable_op.getResults()); return success(); } // Abstracts the conversion of the embedded lookup composite function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
SmallVector<Type> arg_types; SmallVector<Location> arg_locs; for (const Value arg : xla_call_module_op.getArgs()) { arg_types.push_back(arg.getType()); arg_locs.push_back(arg.getLoc()); } SmallVector<Type> output_types; for (const Value output : xla_call_module_op.getOutput()) { output_types.push_back(output.getType()); } entry_func_op.setFunctionType(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
if (existing) return *existing; } ModuleOp module = region.getParentOfType<ModuleOp>(); OpBuilder builder(module.getContext()); auto loc = region.getParentOp()->getLoc(); Block& entry = region.front(); int num_region_arguments = entry.getNumArguments(); Operation* terminator = entry.getTerminator(); // Build the function type. Region arguments and extern values together
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc
builder_.setInsertionPointAfter(op); const Value original_result = op->getResult(result_index); QuantizeValue(original_result, quantized_type, op->getLoc()); } void QuantizationDriver::QuantizeArg(BlockArgument arg, const QuantizedType quantized_type) { builder_.setInsertionPointToStart(arg.getOwner());
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/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc
indices_type = RankedTensorType::get(new_start_indices_shape, indices_type.getElementType()); indices = rewriter.create<mhlo::ReshapeOp>(parent_op->getLoc(), indices_type, indices); } else if (index_vector_dim != indices_type.getRank() - 1) { // If index_vector_dim isn't the last dimension in indices then it isn't
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0)