Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.h29 StatusOr<HloInstruction*> PartitionConvolution(
Dconvolution_handler.cc964 StatusOr<HloInstruction*> PartitionConvolution( in PartitionConvolution() function
Ddot_handler.cc2525 PartitionConvolution(lhs, rhs, output_base_shape, output_sharding, in PartitionDot()