Searched defs:operand_dim (Results 1 – 9 of 9) sorted by relevance
334 int64 operand_dim; member434 int64_t operand_dim) { in MapPassthroughOperandDimToResultDim()694 auto is_reshape_passthrough_operand_dim = [&](int64_t operand_dim) { in FoldReshapeOfGatherNoDegenerateDims()
381 auto operand_dim = GetOperandConcatDim( in ModifyHloPropertiesForConcatShape() local410 auto operand_dim = GetOperandConcatDim( in ModifyHloPropertiesForConcatShape() local
947 const int64_t operand_dim = in GatherParallelDataOperandSharding() local1349 auto operand_dim = GetDimensionForIota(maybe_iota->operand(0)); in GetDimensionForIota() local
1184 int64_t operand_dim = instruction->dimensions(output_dim); in ChooseOperandLayoutFromOutputLayout() local1245 int64_t operand_dim = LayoutUtil::Minor(operand_layout, i); in ChooseOutputLayoutFromOperandLayout() local
2157 int64 operand_dim = dim_numbers.start_index_map(dim); in EmitElementalGather() local
1354 int64_t operand_dim = in ReductionPreservesLayout() local
3521 int64_t operand_dim = 0; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
4117 uint64_t operand_dim = scatter_dimension_numbers() in fold() local
2522 int64_t operand_dim = in EmitScatter() local