Home
last modified time | relevance | path

Searched refs:allow_cpu_custom_calls (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla.cc75 compiler_options.allow_cpu_custom_calls = true; in ConvertGraphToXla()
Dxla_compiler.h165 bool allow_cpu_custom_calls = false; member
/external/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.cc143 options.allow_cpu_custom_calls = in GenerateCompilerOptions()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfake_quantize_ops.cc138 if (ctx->compiler()->options().allow_cpu_custom_calls && in Compile()
246 if (ctx->compiler()->options().allow_cpu_custom_calls && in Compile()
Dsoftmax_op.cc73 if (ctx->compiler()->options().allow_cpu_custom_calls && in Compile()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc292 compiler_options.allow_cpu_custom_calls = false; in CompileTFFunctionToHlo()