- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 89 for getStringAttr (0.78 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
auto variableOp = globalBuilder.create<ml_program::GlobalOp>( globalTensor.getLoc(), name, globalTensor.getType(), globalTensor.getIsMutable(), initial_value, /*visibility=*/globalBuilder.getStringAttr("private")); variableOp.setPrivate(); } SymbolTable syms(module); for (auto func : module.getOps<func::FuncOp>()) { if (!tf_saved_model::IsExported(func)) { continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
Block* insertion_point) { rewriter.setInsertionPointToStart(insertion_point); auto ftype = FunctionType::get(rewriter.getContext(), {type}, {type}); auto name = rewriter.getStringAttr("gelu_decomp"); func::FuncOp new_func = rewriter.create<func::FuncOp>( insertion_point->front().getLoc(), name, ftype); new_func.setPrivate(); new_func.addEntryBlock();
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/lite/stablehlo/transforms/tfl_stablehlo_pass.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/parallel_execute_to_islands.cc
",", group_annotation); } for (auto& op : execute_island.GetBody()) { op.setAttr(TF::kParallelExecAnnotation, builder->getStringAttr(group_annotation)); } } executes.push_back(execute_island); } } void CreateIslandsFromParallelExecute( tf_executor::IslandOp island_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 19 19:47:16 UTC 2023 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc
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/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc
llvm::zip_equal(quantizable_values, aggregator_ids)) { // ID attribute will have empty value for now. SmallVector<NamedAttribute, 5> attributes{ rewriter.getNamedAttr("id", rewriter.getStringAttr(aggregator_id)), rewriter.getNamedAttr( "calibration_method", rewriter.getI32IntegerAttr(calib_opts_.calibration_method())), rewriter.getNamedAttr("num_bins",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
CreateEinsumEquation(dot_dimension_numbers, lhs_rank, rhs_rank); return builder.create<TF::EinsumOp>(loc, output.getType(), input_arguments, builder.getStringAttr(einsum_equation)); } // Restores the collapsed dimensions to the `tensor_type`. `collapsed_dims` // designate the dimension indices that were collapsed to produce `tensor_type`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
output_types = {output_type.clone(elem_type.getStorageType())}; } SmallVector<Value> args = {q_op.getArg(), scale, zero_point}; FlatSymbolRefAttr func_name = FlatSymbolRefAttr::get(rewriter.getStringAttr(kQuantizeFuncName)); auto quantize_call = rewriter.create<TF::PartitionedCallOp>( loc, output_types, args, func_name, /*config=*/"", /*config_proto=*/"", /*executor_type=*/"");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
input_cell_state_, input_layer_norm_coefficients_, forget_layer_norm_coefficients_, cell_layer_norm_coefficients_, output_layer_norm_coefficients_, builder_.getStringAttr("TANH"), builder_.getF32FloatAttr(10.0), builder_.getF32FloatAttr(0.0), mlir::TFL::LSTMKernelTypeAttr::get(builder_.getContext(), mlir::TFL::LSTMKernelType::FULL),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize.td
// MatrixSetDiag op patterns. //===----------------------------------------------------------------------===// class GetStrAttr<string x>: NativeCodeCall< "$_builder.getStringAttr(\"" # x # "\")">; def MatrixSetDiagToV3 : Pat< (TF_MatrixSetDiagOp:$src $input, $diag), (TF_MatrixSetDiagV3Op:$dest $input, $diag, (TF_ConstOp (GetI32Attr<0>)), (GetStrAttr<"RIGHT_LEFT">)),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 06 18:42:28 UTC 2023 - 17K bytes - Viewed (0)