Searched refs:GetI64DenseElementsAttr (Results 1 – 2 of 2) sorted by relevance
162 DenseIntElementsAttr GetI64DenseElementsAttr(const T& container) { in GetI64DenseElementsAttr() function175 return GetI64DenseElementsAttr(elements); in GetWindowElements()
550 GetI64DenseElementsAttr(xla_scatter_dim.update_window_dims()), in GetScatterDimensionNumbers()551 GetI64DenseElementsAttr(xla_scatter_dim.inserted_window_dims()), in GetScatterDimensionNumbers()552 GetI64DenseElementsAttr(xla_scatter_dim.scatter_dims_to_operand_dims()), in GetScatterDimensionNumbers()669 GetI64DenseElementsAttr(hlo_dims.lhs_batch_dimensions()), in EmitGemm()670 GetI64DenseElementsAttr(hlo_dims.rhs_batch_dimensions()), in EmitGemm()671 GetI64DenseElementsAttr(hlo_dims.lhs_contracting_dimensions()), in EmitGemm()672 GetI64DenseElementsAttr(hlo_dims.rhs_contracting_dimensions()), in EmitGemm()952 reduce_op.dimensionsAttr(GetI64DenseElementsAttr(dimensions)); in EmitReduceOp()964 map_op.dimensionsAttr(GetI64DenseElementsAttr(dimensions)); in EmitMapOp()1293 fft.fft_lengthAttr(GetI64DenseElementsAttr(instr->fft_length())); in EmitFftOp()