Home
last modified time | relevance | path

Searched refs:MakePadHlo (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_padding_legalization.cc85 input = MakePadHlo(input, padding, padding_config).ValueOrDie(); in MaybePaddedAndSlicedInput()
145 return MakePadHlo(kernel, padding, padding_config).ValueOrDie(); in MaybePaddedKernel()
257 MakePadHlo(input, padding, input_padding_config).ValueOrDie(); in CanonicalizeBackwardFilterConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.h44 StatusOr<HloInstruction*> MakePadHlo(HloInstruction* operand,
Dhlo_creation_utils.cc57 StatusOr<HloInstruction*> MakePadHlo(HloInstruction* operand, in MakePadHlo() function
434 return MakePadHlo(operand, zero, padding_config); in PadVectorWithZeros()
Dalgebraic_simplifier.cc2265 MakePadHlo(pad->mutable_operand(0), in HandlePad()