Home
last modified time | relevance | path

Searched defs:operand_no (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment.cc106 int64_t operand_no, bool mandatory, bool dfs, int64_t priority) in OperandLayoutConstraint()
237 int64_t operand_no, bool mandatory, in SetOperandLayout()
300 const HloInstruction* instruction, int64_t operand_no, in InsertOperandLayoutConstraint()
331 const Layout& layout, const HloInstruction* instruction, int64_t operand_no, in SetArrayOperandLayout()
1051 int64_t operand_no) { in CopyOperandIfLayoutsDiffer()
1259 int64_t operand_no) { in ChooseOperandLayoutFromOutputLayout()
1335 int64_t operand_no) { in ChooseOutputLayoutFromOperandLayout()
1470 for (int64_t operand_no : in GetArrayUsesOfBuffer() local
1595 for (int64_t operand_no = 0; operand_no < user->operand_count(); in PropagateOperandConstraint() local
1608 for (int64_t operand_no = 0; operand_no < user->operand_count(); in PropagateOperandConstraint() local
[all …]
H A Dlayout_assignment.h116 const int64_t operand_no() const { return operand_no_; } in operand_no() function
H A Dlayout_assignment_test.cc491 for (int64_t operand_no = 0; operand_no < instruction->operand_count(); in PropagateBufferConstraint() local
H A Dhlo_dataflow_analysis.cc1958 int64_t operand_no = user->opcode() == HloOpcode::kTriangularSolve ? 1 : 0; in CanShareOperandBufferWithUser() local
H A Dhlo_verifier.cc746 int64_t operand_no) { in CheckIsTokenOperand()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_layout_assignment.cc130 for (int64_t operand_no = 0; operand_no < instruction->operand_count(); in AddBackendConstraints() local
/aosp_15_r20/frameworks/libs/native_bridge_support/android_api/
Dapi_analysis.py138 self, operand_no, guest_name, host_name, name_pair): argument
/aosp_15_r20/frameworks/libs/binary_translation/kernel_api/tools/
Dapi_analysis.py136 self, operand_no, guest_name, host_name, name_pair): argument