Searched refs:padded_operand (Results 1 – 4 of 4) 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()
1839 TF_ASSIGN_OR_RETURN(HloInstruction * padded_operand, in Propagate()1845 padded_operand)); in Propagate()
1982 const HloInstruction* padded_operand() const { return operand(0); } in padded_operand() function
4246 auto* padded_operand = in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local4253 {padded_operand})); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()