Home
last modified time | relevance | path

Searched refs:set_replace_transpose_with_bitcast (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Damdgpu_compiler.cc100 options.set_replace_transpose_with_bitcast(false); in OptimizeHloConvolutionCanonicalization()
Dgpu_compiler.cc340 options.set_replace_transpose_with_bitcast(false); in OptimizeHloModule()
409 options.set_replace_transpose_with_bitcast(false); in OptimizeHloModule()
562 options.set_replace_transpose_with_bitcast(false); in OptimizeHloPostLayoutAssignment()
Dnvptx_compiler.cc145 options.set_replace_transpose_with_bitcast(false); in OptimizeHloConvolutionCanonicalization()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.h150 void set_replace_transpose_with_bitcast(bool replace_transpose_with_bitcast) { in set_replace_transpose_with_bitcast() function
Dalgebraic_simplifier_test.cc2724 options.set_replace_transpose_with_bitcast(false); in TEST_F()
2729 options.set_replace_transpose_with_bitcast(true); in TEST_F()