Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3526 DimensionVector input_dims_to_keep_minor_to_major = in GetReductionToVectorDimensions() local
3529 input_shape.layout(), input_dims_to_keep_minor_to_major)); in GetReductionToVectorDimensions()
3531 if (input_dims_to_keep_minor_to_major.empty()) { in GetReductionToVectorDimensions()
3542 input_dims_to_keep_minor_to_major.back())) { in GetReductionToVectorDimensions()
3546 input_dims_to_keep_minor_to_major.front())) { in GetReductionToVectorDimensions()