Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_gather_thunk.cc41 auto operands_are_supported = [hlo]() { in CanImplement() local
48 operands_are_supported(); in CanImplement()
52 bool operands_are_supported = in CanImplement() local
58 return op.all_gather_dimension() == 0 && operands_are_supported; in CanImplement()
Dnccl_all_to_all_thunk.cc44 bool operands_are_supported = in CanImplement() local
50 return op.split_dimension().getValueOr(0) == 0 && operands_are_supported; in CanImplement()
Dnccl_all_reduce_thunk.cc101 bool operands_are_supported = in CanImplement() local
107 return operands_are_supported && MatchReductionComputation(op).has_value(); in CanImplement()