Home
last modified time | relevance | path

Searched refs:CheckIsTokenOperand (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.h178 Status CheckIsTokenOperand(const HloInstruction* instruction,
Dhlo_verifier.cc395 Status ShapeVerifier::CheckIsTokenOperand(const HloInstruction* instruction, in CheckIsTokenOperand() function in xla::ShapeVerifier
423 TF_RETURN_IF_ERROR(CheckIsTokenOperand(instruction, 0)); in HandleInfeed()
433 TF_RETURN_IF_ERROR(CheckIsTokenOperand(instruction, 1)); in HandleOutfeed()
1157 TF_RETURN_IF_ERROR(CheckIsTokenOperand(add_dependency, 1)); in HandleAddDependency()