Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for MlirToHloArgs (0.2 sec)

  1. tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph_test.cc

      module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}} {
      func.func @main() -> () {
        func.return
      }
    })";
    
    MlirToHloArgs CreateTestMlirToHloArgs(const char* module_str = kMlirModuleStr) {
      MlirToHloArgs mlir_to_hlo_args;
      mlir_to_hlo_args.rollout_state =
          ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_DISABLED;
      mlir_to_hlo_args.mlir_module = module_str;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 08:08:57 UTC 2024
    - 11.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.cc

    #include "tsl/platform/statusor.h"
    
    namespace tensorflow {
    namespace tf2xla {
    namespace v1 {
    
    using ::tensorflow::tpu::FunctionToHloArgs;
    using ::tensorflow::tpu::GuaranteedConsts;
    using ::tensorflow::tpu::MlirToHloArgs;
    using ::tensorflow::tpu::ShardingAndIndex;
    
    auto* phase2_bridge_compilation_status =
        tensorflow::monitoring::Counter<1>::New(
            "/tensorflow/core/tf2xla/api/v1/"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 14K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc

    namespace tensorflow {
    namespace tf2xla {
    namespace v2 {
    
    using ::tensorflow::monitoring::testing::CellReader;
    using ::testing::Not;
    using ::testing::TestWithParam;
    using tpu::FunctionToHloArgs;
    using tpu::MlirToHloArgs;
    using tpu::ShardingAndIndex;
    using tpu::TPUCompileMetadataProto;
    
    static constexpr char kCompilationTimeStreamzName[] =
        "/tensorflow/core/tf2xla/api/v2/phase2_compilation_time";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 13 23:59:33 UTC 2024
    - 16.1K bytes
    - Viewed (0)
Back to top