Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_gather_thunk.cc44 IsTypeSupportedByNccl(operand->shape().element_type()); in CanImplement()
56 IsTypeSupportedByNccl(shape.element_type()); in CanImplement()
Dnccl_collective_thunk.cc114 bool IsTypeSupportedByNccl(PrimitiveType element_type) { in IsTypeSupportedByNccl() function
Dnccl_collective_thunk.h133 bool IsTypeSupportedByNccl(PrimitiveType element_type);
Dnccl_all_to_all_thunk.cc48 IsTypeSupportedByNccl(shape.element_type()); in CanImplement()
Dnccl_all_reduce_thunk.cc105 IsTypeSupportedByNccl(shape.element_type()); in CanImplement()