Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for tf_xla_enable_xla_devices (0.66 sec)

  1. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

    #include "tensorflow/core/platform/test.h"
    
    using ::tensorflow::testing::FindNodeByName;
    
    namespace tensorflow {
    namespace {
    
    static bool Initialized = [] {
      tensorflow::GetXlaDeviceFlags()->tf_xla_enable_xla_devices = true;
      return true;
    }();
    
    REGISTER_OP("UncompilableNullary").Output("o: float");
    REGISTER_OP("UncompilableUnary").Input("a: float").Output("o: float");
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
Back to top