| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_execution_profile.cc | 35 InsertOrDie(&computation_to_profile_idx_, computation, in HloProfileIndexMap()
40 InsertOrDie(&instruction_to_profile_idx_, instruction, in HloProfileIndexMap()
45 InsertOrDie(&extra_metric_to_profile_idx_, key, current_profile_index++); in HloProfileIndexMap()
|
| D | while_loop_invariant_code_motion.cc | 78 InsertOrDie(hoisted_instructions, old_instruction, new_instruction); in CreateLoopInvariantCopy()
174 InsertOrDie(&unhoisted_invariant_instructions, instr); in TryHoistingInvariantInstructionsFromWhileBody()
259 InsertOrDie(&unhoisted_invariant_instructions, instruction); in TryHoistingInvariantInstructionsFromWhileBody()
|
| D | multi_output_fusion.cc | 40 InsertOrDie(&candidates_index_, it, index++); in Run()
163 InsertOrDie(&candidates_index_, use, index++); in Update()
|
| D | hlo_schedule.cc | 118 InsertOrDie(&id_to_instruction, instruction->unique_id(), instruction); in UpdateComputationSchedule()
124 InsertOrDie(&ids_in_schedule, id); in UpdateComputationSchedule()
|
| D | instruction_fusion.cc | 333 InsertOrDie(&post_order_index_, post_order_[i], i); in ReversePostOrderFusionQueue()
427 InsertOrDie(&post_order_index_, fusion, post_order_.size()); in OnFusingInstruction()
|
| D | hlo_ordering.cc | 365 InsertOrDie(&order_position_, order[i], i); in Initialize()
|
| D | indexed_array_analysis.cc | 101 InsertOrDie(&dfs_state_map, root, kDiscovered); in TraverseAndPopulateCache()
127 InsertOrDie(&cache_, instr, array); in TraverseAndPopulateCache()
|
| D | hlo_module.cc | 446 InsertOrDie(&outlined_instructions, instruction_to_outline, in OutlineExpressionFromComputation()
|
| D | hlo_instructions.cc | 1285 InsertOrDie(&old_to_new, fused_instruction, in MergeFusionInstructionIntoMultiOutput()
1296 InsertOrDie(&old_to_new, fused_instruction, cloned_instruction); in MergeFusionInstructionIntoMultiOutput()
|
| D | buffer_assignment.cc | 1293 InsertOrDie(&live_buffers, buffer); in ComputePeakMemoryLogicalBuffers()
|
| D | layout_assignment.cc | 1669 InsertOrDie(&computation_layouts_, computation, computation_layout); in CalculateComputationLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | map_util.h | 90 void InsertOrDie(Collection* const collection, Key&& key, Value&& value) { in InsertOrDie() function
104 void InsertOrDie(Set* const set, Value&& value) { in InsertOrDie() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk_schedule.cc | 58 InsertOrDie(&hlo_to_thunk, thunk->hlo_instruction(), thunk.get()); in ThunkSchedule()
88 InsertOrDie(&thunk_to_total_order, thunk_total_order_[i], i); in RemoveRedundantDependencyEdges()
|
| D | stream_assignment.cc | 42 InsertOrDie(&hlo_to_stream_number_, hlo, stream_num); in AssignStreamToHlo()
|
| D | buffer_allocations.cc | 37 InsertOrDie(®istered_buffers_, index, address); in RegisterBuffer()
|
| D | hlo_to_ir_bindings.cc | 186 InsertOrDie(&base_ptrs_, &hlo, ShapeTree<llvm::Value*>(hlo_shape, nullptr)); in BindHloToIrValue()
|
| D | gpu_executable.cc | 217 InsertOrDie(&globals, i, global); in ResolveConstantGlobals()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.h | 306 InsertOrDie(&metadata_, kind, md); in AddMetadata()
|
| /external/protobuf/src/google/protobuf/stubs/ |
| D | map_util.h | 338 void InsertOrDie(Collection* const collection,
353 void InsertOrDie(Collection* const collection,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 120 InsertOrDie(parsed_results, parsed_line.opcode, parsed_line);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 143 InsertOrDie(&emitted_functions_, computation, ir_function); in EmitComputation()
204 InsertOrDie(&emitted_literals_, &literal, global_for_const); in EmitConstantGlobals()
207 InsertOrDie(&constant_buffer_to_global_, allocation.index(), in EmitConstantGlobals()
1378 InsertOrDie(&unreduced_dim_map, i, i - delta); in ReductionPreservesLayout()
|