HomeSort by relevance Sort by last modified time
    Searched refs:CollectiveImplementationInterface (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
collective_nccl_reducer.h 23 class NcclReducer : public CollectiveImplementationInterface {
  /external/tensorflow/tensorflow/core/framework/
collective.h 366 class CollectiveImplementationInterface {
368 virtual ~CollectiveImplementationInterface() = default;
405 using Factory = std::function<CollectiveImplementationInterface*()>;
410 CollectiveImplementationInterface** implementation);
418 CollectiveImplementationInterface** implementation);
422 std::vector<CollectiveImplementationInterface*>* implementations);
434 CollectiveImplementationInterface** implementation,
collective.cc 40 CollectiveImplementationInterface* param_resolver_instance;
181 CollectiveImplementationInterface** implementation) {
188 CollectiveImplementationInterface** implementation) {
194 std::vector<CollectiveImplementationInterface*>* implementations) {
216 CollectiveImplementationInterface** implementation, bool param_resolver) {
  /external/tensorflow/tensorflow/core/common_runtime/
hierarchical_tree_broadcaster.h 26 class HierarchicalTreeBroadcaster : public CollectiveImplementationInterface {
ring_alg.h 31 class RingAlg : public CollectiveImplementationInterface {
base_collective_executor.h 167 CollectiveImplementationInterface** col_impl);
base_collective_executor.cc 246 CollectiveImplementationInterface* col_impl = nullptr;
285 CollectiveImplementationInterface** col_impl) {
collective_param_resolver_local.cc 671 CollectiveImplementationInterface* col_impl;

Completed in 145 milliseconds