Home
last modified time | relevance | path

Searched refs:CastOrNull (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_casting_utils.h58 const T* CastOrNull(const HloInstruction* instruction) { in CastOrNull() function
64 T* CastOrNull(HloInstruction* instruction) { in CastOrNull() function
66 CastOrNull<T>(const_cast<const HloInstruction*>(instruction))); in CastOrNull()
Dhlo_casting_utils_test.cc69 DummyInstruction* casted = CastOrNull<DummyInstruction>(null); in TEST()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc146 CastOrNull<HloBroadcastInstruction>(bias_broadcast_instr); in FindConvWithRelu()
168 CastOrNull<HloConstantInstruction>(alpha_conv_instr), in FindConvWithRelu()
169 CastOrNull<HloConstantInstruction>(alpha_side_input_instr)}; in FindConvWithRelu()