Searched refs:operand_dim (Results 1 – 8 of 8) sorted by relevance
334 int64 operand_dim; member399 value.operand_dim); in ComputeReshapePassthroughDimPairs()414 return lhs.operand_dim < rhs.operand_dim; in ComputeReshapePassthroughDimPairs()426 return passthrough_dim_pair.operand_dim == dim; in IsReshapePassthroughOperandDim()434 int64 operand_dim) { in MapPassthroughOperandDimToResultDim() argument437 return passthrough_dim_pair.operand_dim == operand_dim; in MapPassthroughOperandDimToResultDim()694 auto is_reshape_passthrough_operand_dim = [&](int64 operand_dim) { in FoldReshapeOfGatherNoDegenerateDims() argument696 operand_dim); in FoldReshapeOfGatherNoDegenerateDims()
1099 int64 operand_dim = instruction->dimensions(output_dim); in ChooseOperandLayoutFromOutputLayout() local1100 new_minor_to_major[i] = operand_dim; in ChooseOperandLayoutFromOutputLayout()1160 int64 operand_dim = LayoutUtil::Minor(operand_layout, i); in ChooseOutputLayoutFromOperandLayout() local1161 int64 user_dim = inverse_dimensions[operand_dim]; in ChooseOutputLayoutFromOperandLayout()
1971 int64 operand_dim = dim_numbers.start_index_map(dim); in EmitElementalGather() local1972 int64 output_dim = operand_to_output_dim[operand_dim]; in EmitElementalGather()1979 operand_shape.dimensions(operand_dim) - output_dim_size; in EmitElementalGather()1996 operand_multi_index[operand_dim] = in EmitElementalGather()1997 Add(operand_multi_index[operand_dim], maybe_truncated_clamped_index); in EmitElementalGather()
913 const int64 operand_dim = in GatherParallelDataOperandSharding() local915 operand_tile_assignment[operand_dim] = in GatherParallelDataOperandSharding()
3148 int64 operand_dim = 0; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local3151 if (absl::c_linear_search(out->dimensions(), operand_dim)) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3153 ++operand_dim; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3158 padded_shape.dimensions(operand_dim)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3159 ++operand_dim; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
1352 int64 operand_dim = operand_shape.layout().minor_to_major(operand_dim_idx); in ReductionPreservesLayout() local1353 if (!reduced_dims.contains(operand_dim)) { in ReductionPreservesLayout()1354 if (FindOrDie(unreduced_dim_map, operand_dim) != in ReductionPreservesLayout()
2668 int64 operand_dim = in EmitScatter() local2679 Add(input_window_multidim[operand_dim], casted_scatter_index); in EmitScatter()2680 input_window_multidim[operand_dim] = dim_offset; in EmitScatter()2683 int64 max_index = desc.operand_shape.dimensions(operand_dim) - in EmitScatter()2684 input_window_bounds[operand_dim] + 1; in EmitScatter()
3045 uint64_t operand_dim = scatter_dimension_numbers() in fold() local3050 base_index[operand_dim] += in fold()