Searched defs:operand_dim (Results 1 – 5 of 5) sorted by relevance
334 int64 operand_dim; member434 int64 operand_dim) { in MapPassthroughOperandDimToResultDim()694 auto is_reshape_passthrough_operand_dim = [&](int64 operand_dim) { in FoldReshapeOfGatherNoDegenerateDims()
1041 int64 operand_dim = instruction->dimensions(output_dim); in ChooseOperandLayoutFromOutputLayout() local1102 int64 operand_dim = LayoutUtil::Minor(operand_layout, i); in ChooseOutputLayoutFromOperandLayout() local
1915 int64 operand_dim = dim_numbers.start_index_map(dim); in EmitElementalGather() local
1387 int64 operand_dim = operand_shape.layout().minor_to_major(operand_dim_idx); in ReductionPreservesLayout() local
1147 int64 operand_dim = dim_numbers.scatter_dims_to_operand_dims(i); in EmitScatter() local