- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for NameLoc (0.44 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc
RankedTensorType::get(/*shape=*/{}, builder.getType<TF::StringType>()); BlockArgument new_file_prefix_arg = main_func_op.getBody().front().addArgument( file_prefix_arg_type, NameLoc::get(builder.getStringAttr(kTfFilePrefix))); SmallVector<Type> input_types(main_func_op.getArgumentTypes()); input_types.emplace_back(file_prefix_arg_type); main_func_op.setType(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/import_quant_stats_pass.cc
Location loc = tensorflow::GetLocationWithoutOpType(op->getLoc()); if (auto name = loc.dyn_cast<NameLoc>()) { return name.getName().strref(); } else if (auto fused_name = loc.dyn_cast<FusedLoc>()) { for (auto sub_loc : fused_name.getLocations()) { if (auto named_sub_loc = sub_loc.dyn_cast<NameLoc>()) { return named_sub_loc.getName().strref(); } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 10:41:08 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
llvm::SmallVector<Value> returning_values(output_count, Value()); for (const auto& node_name : node_to_output_map.keys()) { auto node_output_tensors = node_to_output_map[node_name]; NameLoc new_loc = NameLoc::get(builder.getStringAttr(node_name)); int32_t max_tensor_index = 0; absl::c_for_each(node_output_tensors, [&max_tensor_index](const OutputInfo& output_info) {
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/lite/utils/lstm_utils_test.cc
layer_norm_scale_type}; auto func_type = builder->getFunctionType(input_types, output_type); auto func = func::FuncOp::create( mlir::NameLoc::get(builder->getStringAttr("fused_func")), "fused_func", func_type, {}); func.addEntryBlock(); std::vector<std::string> attributes; if (ln) { attributes.push_back(kLayerNormalizedLstmCellSimple);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
const std::string init_type = GetInitializerType(init_func_ops); const std::string name = absl::StrCat(init_type, "_", init_func_ops.getName().str()); return NameLoc::get(StringAttr::get(ctx, name)); } void MergeInitializerFunctionOpsToMainPass::runOnOperation() { ModuleOp module_op = getOperation(); MLIRContext* ctx = module_op.getContext();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
bool IsOpQuantizable(Operation* op); // Specialized version of location to string for flatbuffer exported locations. inline std::string GetTensorNameFromLoc(Location loc) { if (auto name_loc = loc.dyn_cast<NameLoc>()) { return name_loc.getName().str(); } return ""; } template <typename QuantizeOpT, typename DequantizeOpT> struct ConvertStatsToQDQs : public OpRewritePattern<quantfork::StatisticsOp> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
mlir::Location unwrapped_inst_loc = GetLocationWithoutOpType(inst_loc); if (auto call_site = mlir::dyn_cast<mlir::CallSiteLoc>(unwrapped_inst_loc)) { if (auto name_loc = mlir::dyn_cast<mlir::NameLoc>( GetLocationWithoutOpType(call_site.getCallee()))) { llvm::StringRef original_node_name, original_func_name; std::tie(original_node_name, original_func_name) =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
assert(while_wrapper && "While op is expected to be wrapped in a IslandOp"); // Get the dummy constant. OpBuilder builder(while_wrapper); auto loc = NameLoc::get( builder.getStringAttr("chain_control_outputs@" + while_op.getBody())); IslandOp const_wrapper = GetDummyConstant(builder, const_type, loc); // Get new operand and result types.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0)