Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc79 const Shape& shape, absl::Span<const int64> dims_middle) { in PartitionShapeByMiddleDimensions() argument
80 CHECK(LayoutUtil::AreDimensionsConsecutive(shape.layout(), dims_middle)); in PartitionShapeByMiddleDimensions()
88 bool cur_dim_in_middle = absl::c_linear_search(dims_middle, cur_dim); in PartitionShapeByMiddleDimensions()