Searched refs:operand_in_alternate_mem (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment.h | 123 absl::optional<int64> operand_in_alternate_mem = absl::nullopt,
137 absl::optional<int64> operand_in_alternate_mem,
|
| D | memory_space_assignment.cc | 259 absl::optional<int64> operand_in_alternate_mem, in GetInstructionElapsedDueToMemory() argument
265 if (operand_in_alternate_mem) { in GetInstructionElapsedDueToMemory()
269 instruction, *operand_in_alternate_mem); in GetInstructionElapsedDueToMemory()
302 absl::optional<int64> operand_in_alternate_mem, in GetInstructionElapsedInAlternateMemory() argument
306 GetInstructionElapsedDueToMemory(instruction, operand_in_alternate_mem, in GetInstructionElapsedInAlternateMemory()
|
| D | memory_space_assignment_test.cc | 344 absl::optional<int64> operand_in_alternate_mem, in GetInstructionElapsedInAlternateMemory() argument
348 instruction, operand_in_alternate_mem, output_in_alternate_mem); in GetInstructionElapsedInAlternateMemory()
350 if (operand_in_alternate_mem) { in GetInstructionElapsedInAlternateMemory()
|