Search Options

Results per page
Sort
Preferred Languages
Advance

Results 71 - 80 of 89 for getStringAttr (0.78 sec)

  1. 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)
  2. 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)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc

                attrs.push_back(named_attr);
              } else {
                std::vector<StringRef> temp;
                for (size_t i = 0; i < vector.size(); i++) {
                  auto conf_str =
                      builder->getStringAttr(vector[i].AsString().str());
                  temp.push_back(conf_str);
                }
                ArrayRef<StringRef> values(temp);
                auto named_attr =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 24 06:08:43 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top