Home
last modified time | relevance | path

Searched refs:ConvertDenseIntAttr (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc136 static std::vector<int64> ConvertDenseIntAttr(mlir::DenseIntElementsAttr attr) { in ConvertDenseIntAttr() function
141 static std::vector<int64> ConvertDenseIntAttr( in ConvertDenseIntAttr() function
144 return ConvertDenseIntAttr(*attr); in ConvertDenseIntAttr()
153 return ConvertDenseIntAttr(*broadcast_dimensions); in Convert_broadcast_dimensions()
190 return ConvertDenseIntAttr(attribute); \
297 auto offset_dims = ConvertDenseIntAttr(input.offset_dims()); in Convert_dimension_numbers()
302 auto collapsed_slice_dims = ConvertDenseIntAttr(input.collapsed_slice_dims()); in Convert_dimension_numbers()
307 auto start_index_map = ConvertDenseIntAttr(input.start_index_map()); in Convert_dimension_numbers()
321 auto update_window_dims = ConvertDenseIntAttr(input.update_window_dims()); in Convert_scatter_dimension_numbers()
326 auto inserted_window_dims = ConvertDenseIntAttr(input.inserted_window_dims()); in Convert_scatter_dimension_numbers()
[all …]