Searched defs:operand_num (Results 1 – 11 of 11) sorted by relevance
1002 int64 operand_num, in operand_bytes_accessed()1033 void HloCostAnalysis::SetOperandBytesAccessed(int64 operand_num, float value) { in SetOperandBytesAccessed()1037 void HloCostAnalysis::SetOperandBytesAccessed(int64 operand_num, in SetOperandBytesAccessed()1052 int64 operand_num, ShapeIndex index) { in GetOperandBytesAccessedKey()
126 bool ShouldSkipPadOnOperand(const HloInstruction* inst, int64 operand_num, in ShouldSkipPadOnOperand()1023 for (int64 operand_num = 0; operand_num < inst->operand_count(); in Run() local
481 for (int64 operand_num = 0; in OutlineExpressionFromComputation() local
1639 for (int64 operand_num = 0; operand_num < operand_count(); ++operand_num) { in ~HloInstruction() local2058 Status HloInstruction::ReplaceOperandWith(int64 operand_num, in ReplaceOperandWith()2069 int64 operand_num, HloInstruction* new_operand) { in ReplaceOperandWithDifferentShape()
1194 int64 operand_num, bool control_edge = false) { in AddInstructionIncomingEdges()
1539 for (int64 operand_num = 0; operand_num < operand_count(); ++operand_num) { in CloneAndFuseInternal() local1572 for (int64 operand_num = 0; operand_num < clone->operand_count(); in CloneAndFuseInternal() local
3264 int64 operand_num = 0; in TrySimplifyScalarSlice() local
279 for (int64 operand_num = 0; operand_num < consumer->operand_count(); in IsFusedIrEmitterInefficient() local
640 int operand_num = use.getOperandNumber(); in PreprocessConstantOps() local
777 if (auto operand_num = in ProfitableToMakeDotOperandColumnMajor() local
1577 int64 operand_num = 0; in CustomCall() local