Home
last modified time | relevance | path

Searched defs:new_output_dims (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dindexed_array_analysis.cc504 DimensionVector new_output_dims; in ReshapeToRemoveDegenerateDims() local
536 DimensionVector new_output_dims = [&]() { in ReshapeToAddDegenerateDims() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_vectorize_convolutions.cc340 DimensionVector new_output_dims(output_shape.dimensions().begin(), in TryRevectorizeConv() local