Home
last modified time | relevance | path

Searched refs:applyTensorflowAndCLOptions (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dkernel_creator.cc118 if (apply_cl_options) applyTensorflowAndCLOptions(pm); in LowerTFToJITInvocation()
146 if (apply_cl_options) applyTensorflowAndCLOptions(pm); in LowerTFtoLoops()
238 if (apply_cl_options) applyTensorflowAndCLOptions(pm); in LowerLoopsToGPU()
333 if (apply_cl_options) tensorflow::applyTensorflowAndCLOptions(pm); in LowerKernelBodiesToLowLevelIr()
356 if (apply_cl_options) applyTensorflowAndCLOptions(pm); in AmendKernelLLVMIRWithStaticKnowledge()
375 if (apply_cl_options) applyTensorflowAndCLOptions(pm); in GenerateDeviceCode()
392 if (apply_cl_options) applyTensorflowAndCLOptions(pm); in LowerHostSideToFinalForm()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
Dpass_utils.cc36 tensorflow::applyTensorflowAndCLOptions(pm); in ConvertLmhloToJitRt()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgraph_optimization_pass.cc48 ::tensorflow::applyTensorflowAndCLOptions(pm); in Run()
Dbridge.cc57 ::tensorflow::applyTensorflowAndCLOptions(bridge); in RunTFXLABridge()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddump_mlir_util.h112 void applyTensorflowAndCLOptions(mlir::PassManager& pm,
Ddump_mlir_util.cc330 void applyTensorflowAndCLOptions(mlir::PassManager& pm, in applyTensorflowAndCLOptions() function
Dcompile_mlir_util.cc465 applyTensorflowAndCLOptions(tf2xla); in LegalizeToHlo()
779 applyTensorflowAndCLOptions(pm); in CompileGraphSetup()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/function/
Dfunction.cc53 tensorflow::applyTensorflowAndCLOptions(pm); in CompileTFMLIRToBEF()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc526 ::tensorflow::applyTensorflowAndCLOptions(pm); in GetFunctionDef()