Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for CompileAheadOfTime (0.14 sec)

  1. tensorflow/compiler/jit/xla_device_compiler_client.cc

          options, result, client_->default_device_ordinal());
      TF_ASSIGN_OR_RETURN(
          std::vector<std::unique_ptr<xla::AotCompilationResult>> aot_results,
          client_->CompileAheadOfTime(*result.computation, argument_layouts,
                                      build_options));
      TF_RET_CHECK(aot_results.size() == 1);
      return aot_results[0]->SerializeAsString();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 4.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/aot/compile.cc

      xla::Shape result_shape(pshape->result());
      instance.result_layout = &result_shape;
      absl::StatusOr<std::vector<std::unique_ptr<xla::AotCompilationResult>>>
          aot_or = client->CompileAheadOfTime({instance}, aot_opts);
      if (!aot_or.ok()) {
        return errors::Unknown("XLA compilation failed: ",
                               aot_or.status().message());
      }
      compile_result->aot =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 08:28:57 UTC 2024
    - 11.9K bytes
    - Viewed (0)
Back to top