Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc3147 auto padded_out_shape = out->shape(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
3150 while (output_dim < padded_out_shape.rank()) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3157 padded_out_shape.set_dimensions(output_dim, in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3164 padded_out_shape, in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()