Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc3276 for (int64 collapsed_dim : dim_numbers.collapsed_slice_dims()) { in ValidateGatherDimensionNumbers() local
3277 if (collapsed_dim < 0 || collapsed_dim >= input_shape.dimensions_size()) { in ValidateGatherDimensionNumbers()
3281 input_shape.dimensions_size(), collapsed_dim); in ValidateGatherDimensionNumbers()
Ddynamic_dimension_inference.cc1206 for (int64 collapsed_dim : gather_dims.collapsed_slice_dims()) { in HandleGather() local
1207 if (collapsed_dim < input_dynamic_dimension) { in HandleGather()