Searched defs:operand_num (Results 1 – 11 of 11) sorted by relevance
1084 int64 operand_num, in operand_bytes_accessed()1151 void HloCostAnalysis::SetOperandBytesAccessed(int64 operand_num, float value) { in SetOperandBytesAccessed()1155 void HloCostAnalysis::SetOperandBytesAccessed(int64 operand_num, in SetOperandBytesAccessed()1170 int64 operand_num, ShapeIndex index) { in GetOperandBytesAccessedKey()
313 for (int64 operand_num = 0; operand_num < old_root->operand_count(); in FindSpecialConverts() local441 for (int64 operand_num = 0; operand_num < old_root->operand_count(); in ConvertSpecialMove() local446 for (int64 operand_num = 0; operand_num < old_root->operand_count(); in ConvertSpecialMove() local
526 for (int64 operand_num = 0; in OutlineExpressionFromComputation() local
221 bool ShouldSkipPadOnOperand(const HloInstruction* inst, int64 operand_num, in ShouldSkipPadOnOperand()1885 for (int64 operand_num = 0; operand_num < inst->operand_count(); in Run() local
1846 for (int64 operand_num = 0; operand_num < operand_count(); ++operand_num) { in DetachFromOperandsAndUsers() local2329 Status HloInstruction::ReplaceOperandWith(int64 operand_num, in ReplaceOperandWith()2340 int64 operand_num, HloInstruction* new_operand) { in ReplaceOperandWithDifferentShape()
1214 int64 operand_num, bool control_edge = false) { in AddInstructionIncomingEdges()
1684 for (int64 operand_num = 0; operand_num < operand_count(); ++operand_num) { in CloneAndFuseInternal() local1717 for (int64 operand_num = 0; operand_num < clone->operand_count(); in CloneAndFuseInternal() local
4006 int64 operand_num = 0; in TrySimplifyScalarSlice() local
689 int operand_num = indexed_use.value().second; in PreprocessConstantOps() local
914 if (auto operand_num = in ProfitableToMakeDotOperandColumnMajor() local
1906 int64 operand_num = 0; in CustomCall() local1993 int64 operand_num = 0; in CustomCall() local