Searched refs:padded_operand (Results 1 – 3 of 3) sorted by relevance
164 auto padded_operand = Pad(operand, Zero(b, dtype), padding_config); in AvgPool() local165 auto pooled = ComputeSums(padded_operand, init_value, kernel_size, stride, in AvgPool()
1302 TF_ASSIGN_OR_RETURN(HloInstruction * padded_operand, in Propagate()1308 padded_operand)); in Propagate()
3235 auto padded_operand = PadToShape(outside_to_inside[broadcast->operand(0)], in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local3241 {padded_operand}))); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()