HomeSort by relevance Sort by last modified time
    Searched refs:user_pre_optimization_hook_ (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
llvm_compiler.h 43 CHECK(!user_pre_optimization_hook_)
46 user_pre_optimization_hook_ = hook;
49 void RemovePreOptimizationHook() { user_pre_optimization_hook_ = nullptr; }
79 ModuleHook user_pre_optimization_hook_; member in class:xla::LLVMCompiler
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_compiler.cc 548 if (user_pre_optimization_hook_) {
549 TF_CHECK_OK(user_pre_optimization_hook_(llvm_module));
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_compiler.cc 468 *module, user_pre_optimization_hook_, user_post_optimization_hook_,
    [all...]

Completed in 241 milliseconds