Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_to_all_thunk.cc91 ToNcclDataType(element_type)); in RunNcclCollective()
123 ToNcclDataType(element_type)); in RunNcclCollective()
Dnccl_all_gather_thunk.cc90 ToNcclDataType(config_.config.operand_element_type[i])); in RunNcclCollective()
Dnccl_utils.h51 StatusOr<ncclDataType_t> ToNcclDataType(PrimitiveType element_type);
Dnccl_all_reduce_thunk.cc141 ToNcclDataType(config_.config.operand_element_type[i])); in RunNcclCollective()
Dnccl_utils.cc49 StatusOr<ncclDataType_t> ToNcclDataType(PrimitiveType element_type) { in ToNcclDataType() function