Searched defs:operand_num (Results 1 – 14 of 14) sorted by relevance
1150 int64_t operand_num, in operand_bytes_accessed()1218 void HloCostAnalysis::SetOperandBytesAccessed(int64_t operand_num, in SetOperandBytesAccessed()1223 void HloCostAnalysis::SetOperandBytesAccessed(int64_t operand_num, in SetOperandBytesAccessed()1238 int64_t operand_num, ShapeIndex index) { in GetOperandBytesAccessedKey()
372 for (int64_t operand_num = 0; operand_num < old_root->operand_count(); in FindSpecialConverts() local527 for (int64_t operand_num = 0; operand_num < old_root->operand_count(); in ConvertSpecialMove() local532 for (int64_t operand_num = 0; operand_num < old_root->operand_count(); in ConvertSpecialMove() local
2184 for (int64_t operand_num = 0; operand_num < operand_count(); ++operand_num) { in DetachFromOperandsAndUsers() local2687 Status HloInstruction::ReplaceOperandWith(int64_t operand_num, in ReplaceOperandWith()2698 int64_t operand_num, HloInstruction* new_operand) { in ReplaceOperandWithDifferentShape()3911 int64_t operand_num) { in OperandElementUse()3972 for (int64_t operand_num = 0; operand_num < hlo.operands().size(); in ComputeInternal() local
233 bool ShouldSkipPadOnOperand(const HloInstruction* inst, int64_t operand_num, in ShouldSkipPadOnOperand()2254 for (int64_t operand_num = 0; operand_num < inst->operand_count(); in Run() local
640 for (int64_t operand_num = 0; in OutlineExpressionFromComputation() local
1376 int64_t operand_num, bool control_edge = false) { in AddInstructionIncomingEdges()
1656 for (int64_t operand_num = 0; operand_num < operand_count(); in CloneAndAppendInstructionIntoCalledComputation() local1691 for (int64_t operand_num = 0; operand_num < clone->operand_count(); in CloneAndAppendInstructionIntoCalledComputation() local
413 for (int operand_num = 0; operand_num < instruction.operand_count(); in GetInstructionElapsedDueToMemory() local
4426 int64_t operand_num = 0; in TrySimplifyScalarSlice() local
132 int operand_num = use.getOperandNumber(); in getQuantizableOps() local
293 int operand_num = use.getOperandNumber(); in getQuantizableOps() local
789 int operand_num = indexed_use.value().second; in PreprocessConstantOps() local
1097 if (auto operand_num = in ProfitableToMakeDotOperandColumnMajor() local
1994 int64_t operand_num = 0; in CustomCall() local2102 int64_t operand_num = 0; in CustomCall() local