Searched refs:operand_dimension (Results 1 – 2 of 2) sorted by relevance
703 for (int64 operand_dimension = 0; operand_dimension < operand_shape.rank(); in HandleBroadcast() local704 ++operand_dimension) { in HandleBroadcast()705 int64 output_dimension = broadcast->dimensions()[operand_dimension]; in HandleBroadcast()709 operand_shape.dimensions(operand_dimension))) in HandleBroadcast()
462 int64 operand_dimension, int64 operand_index, in HandleDot()534 auto iter = result_dim_mapping.find(operand_dimension); in HandleDot()