| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_execution_profile.cc | 36 InsertOrDie(&computation_to_profile_idx_, computation, in HloProfileIndexMap()
41 InsertOrDie(&instruction_to_profile_idx_, instruction, in HloProfileIndexMap()
46 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()
260 InsertOrDie(&unhoisted_invariant_instructions, instruction); in TryHoistingInvariantInstructionsFromWhileBody()
|
| D | multi_output_fusion.cc | 44 InsertOrDie(&candidates_index_, it, index++); in Run()
172 InsertOrDie(&candidates_index_, input_fusion, index); in CreateFusion()
260 InsertOrDie(&candidates_index_, use, index++); in UpdateBeforeFuse()
|
| 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 | 358 InsertOrDie(&post_order_index_, post_order_[i], i); in ReversePostOrderFusionQueue()
450 InsertOrDie(&post_order_index_, fusion, post_order_.size()); in OnFusingInstruction()
|
| D | hlo_ordering.cc | 502 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 | 547 InsertOrDie(&outlined_instructions, instruction_to_outline, in OutlineExpressionFromComputation()
|
| D | layout_assignment.cc | 635 InsertOrDie(&conditional_mismatch_, in AddMandatoryConstraints()
1810 InsertOrDie(&computation_layouts_, computation, computation_layout); in CalculateComputationLayout()
|
| D | hlo_instructions.cc | 1542 InsertOrDie(&old_to_new, fused_instruction, in MergeFusionInstructionIntoMultiOutput()
1553 InsertOrDie(&old_to_new, fused_instruction, cloned_instruction); in MergeFusionInstructionIntoMultiOutput()
|
| D | buffer_assignment.cc | 1563 InsertOrDie(&live_values, value); in ComputePeakMemoryLogicalBuffers()
|
| /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 | 63 InsertOrDie(&hlo_to_thunk, thunk_to_hlo_.at(thunk.get()), thunk.get()); in ThunkSchedule()
94 InsertOrDie(&thunk_to_total_order, thunk_total_order_[i], i); in RemoveRedundantDependencyEdges()
|
| D | stream_assignment.cc | 43 InsertOrDie(&hlo_to_stream_number_, hlo, stream_num); in AssignStreamToHlo()
|
| D | hlo_to_ir_bindings.cc | 164 InsertOrDie(&base_ptrs_, &hlo, ShapeTree<llvm::Value*>(hlo_shape, nullptr)); in BindHloToIrValue()
|
| D | gpu_executable.cc | 293 InsertOrDie(&globals, info.allocation_index, global); in ResolveConstantGlobals()
|
| D | gpu_compiler.cc | 635 InsertOrDie(&thunk_to_hlo, thunk.get(), instruction); in CompileModuleToLlvmIrImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.h | 313 InsertOrDie(&metadata_, kind, md); in AddMetadata()
|
| /external/protobuf/src/google/protobuf/stubs/ |
| D | map_util.h | 338 void InsertOrDie(Collection* const collection,
354 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 | 192 InsertOrDie(&emitted_functions_, computation, ir_function); in EmitComputation()
264 InsertOrDie(&emitted_literals_, &literal, global_for_const); in EmitConstantGlobals()
267 InsertOrDie(&constant_buffer_to_global_, allocation.index(), in EmitConstantGlobals()
1343 InsertOrDie(&unreduced_dim_map, i, i - delta); in ReductionPreservesLayout()
|