Searched defs:operand_dim (Results 1 – 9 of 9) sorted by relevance
340 int64_t operand_dim; member440 int64_t operand_dim) { in MapPassthroughOperandDimToResultDim()700 auto is_reshape_passthrough_operand_dim = [&](int64_t operand_dim) { in FoldReshapeOfGatherNoDegenerateDims()
383 auto operand_dim = GetOperandConcatDim( in ModifyHloPropertiesForConcatShape() local414 auto operand_dim = GetOperandConcatDim( in ModifyHloPropertiesForConcatShape() local
1312 int64_t operand_dim = instruction->dimensions(output_dim); in ChooseOperandLayoutFromOutputLayout() local1389 int64_t operand_dim = LayoutUtil::Minor(operand_layout, i); in ChooseOutputLayoutFromOperandLayout() local
1031 const int64_t operand_dim = in GatherParallelDataOperandSharding() local1472 auto operand_dim = GetDimensionForIota(maybe_iota->operand(0)); in GetDimensionForIota() local
2164 int64_t operand_dim = dim_numbers.start_index_map(dim); in EmitElementalGather() local
1078 int64_t operand_pos = -1, operand_dim = -1; in ComputeBitcastDimMap() local
1445 int64_t operand_dim = in ReductionPreservesLayout() local
4127 int64_t operand_dim = 0; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
2382 int64_t operand_dim = desc.dim_numbers.getScatterDimsToOperandDims()[i]; in EmitScatter() local