Searched refs:CheckOperandCount (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 60 Status CheckOperandCount(const HloInstruction* hlo, int expected) { in CheckOperandCount() function
89 TF_RETURN_IF_ERROR(CheckOperandCount(hlo, *arity)); in Preprocess()
178 TF_RETURN_IF_ERROR(CheckOperandCount(hlo, 1)); in HandleCholesky()
458 TF_RETURN_IF_ERROR(CheckOperandCount(instruction, 2)); in HandleRng()
524 TF_RETURN_IF_ERROR(CheckOperandCount(instruction, 0)); in HandleRngGetAndUpdateState()
964 TF_RETURN_IF_ERROR(CheckOperandCount(conditional, num_branches + 1)); in HandleConditional()
|
| /external/llvm-project/mlir/lib/Rewrite/ |
| D | ByteCode.cpp | 78 CheckOperandCount, enumerator
541 writer.append(OpCode::CheckOperandCount, op.operation(), op.count(), in generate()
899 case CheckOperandCount: { in execute()
|