Searched refs:operand_dim (Results 1 – 5 of 5) 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()
1041 int64 operand_dim = instruction->dimensions(output_dim); in ChooseOperandLayoutFromOutputLayout() local1042 new_minor_to_major[i] = operand_dim; in ChooseOperandLayoutFromOutputLayout()1102 int64 operand_dim = LayoutUtil::Minor(operand_layout, i); in ChooseOutputLayoutFromOperandLayout() local1103 int64 user_dim = inverse_dimensions[operand_dim]; in ChooseOutputLayoutFromOperandLayout()
1915 int64 operand_dim = dim_numbers.start_index_map(dim); in EmitElementalGather() local1916 int64 output_dim = operand_to_output_dim[operand_dim]; in EmitElementalGather()1923 operand_shape.dimensions(operand_dim) - output_dim_size; in EmitElementalGather()1936 operand_multi_index[operand_dim] = in EmitElementalGather()1937 Add(operand_multi_index[operand_dim], in EmitElementalGather()
1147 int64 operand_dim = dim_numbers.scatter_dims_to_operand_dims(i); in EmitScatter() local1157 Add(input_window_multidim[operand_dim], casted_scatter_index); in EmitScatter()1158 input_window_multidim[operand_dim] = dim_offset; in EmitScatter()1161 int64 max_index = operand->shape().dimensions(operand_dim) - in EmitScatter()1162 input_window_bounds[operand_dim] + 1; in EmitScatter()
1387 int64 operand_dim = operand_shape.layout().minor_to_major(operand_dim_idx); in ReductionPreservesLayout() local1388 if (!reduced_dims.contains(operand_dim)) { in ReductionPreservesLayout()1389 if (FindOrDie(unreduced_dim_map, operand_dim) != in ReductionPreservesLayout()