Searched refs:MakePadHlo (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_padding_legalization.cc | 85 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/ |
| D | hlo_creation_utils.h | 44 StatusOr<HloInstruction*> MakePadHlo(HloInstruction* operand,
|
| D | hlo_creation_utils.cc | 57 StatusOr<HloInstruction*> MakePadHlo(HloInstruction* operand, in MakePadHlo() function
434 return MakePadHlo(operand, zero, padding_config); in PadVectorWithZeros()
|
| D | algebraic_simplifier.cc | 2265 MakePadHlo(pad->mutable_operand(0), in HandlePad()
|