Searched defs:operand_dim (Results 1 – 8 of 8) 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()
1099 int64 operand_dim = instruction->dimensions(output_dim); in ChooseOperandLayoutFromOutputLayout() local1160 int64 operand_dim = LayoutUtil::Minor(operand_layout, i); in ChooseOutputLayoutFromOperandLayout() local
913 const int64 operand_dim = in GatherParallelDataOperandSharding() local
1971 int64 operand_dim = dim_numbers.start_index_map(dim); in EmitElementalGather() local
3148 int64 operand_dim = 0; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
1352 int64 operand_dim = operand_shape.layout().minor_to_major(operand_dim_idx); in ReductionPreservesLayout() local
3045 uint64_t operand_dim = scatter_dimension_numbers() in fold() local
2668 int64 operand_dim = in EmitScatter() local