Home
last modified time | relevance | path

Searched refs:FindOrDie (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dthunk_schedule.cc38 depends_on_[&thunk].push_back(FindOrDie(hlo_to_thunk, &operand)); in AddDependenciesOnTransitiveOperands()
120 std::list<const Thunk*>& sources = FindOrDie(depends_on_, dst); in RemoveRedundantDependencyEdges()
126 int src_order = FindOrDie(thunk_to_total_order, src); in RemoveRedundantDependencyEdges()
143 return FindOrDie(depends_on_, thunk); in DependsOn()
Dgpu_executable.cc132 stream->ThenWaitFor(FindOrDie(thunk_to_finish_event, dependency).get()); in ExecuteThunks()
273 buffer_allocations_builder.RegisterBuffer(i, FindOrDie(*globals, i)); in ExecuteOnStream()
Dstream_assignment.cc33 return FindOrDie(hlo_to_stream_number_, &hlo); in StreamNumberForHlo()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_execution_profile.h50 return FindOrDie(instruction_to_profile_idx(), &instruction); in GetProfileIndexFor()
54 return FindOrDie(computation_to_profile_idx(), &computation); in GetProfileIndexFor()
58 return xla::FindOrDie(extra_metric_to_profile_idx(), key); in GetProfileIndexFor()
Dhlo_liveness_analysis.cc125 const ShapeTree<bool>& index_tree = FindOrDie(*live_index_map, instruction); in PropagateLivenessThroughTuple()
155 const ShapeTree<bool>& index_tree = FindOrDie(*live_index_map, instruction); in PropagateLivenessThroughGTE()
174 const ShapeTree<bool>& index_tree = FindOrDie(*live_index_map, instruction); in PropagateLivenessThroughWhile()
206 FindOrDie(*live_index_map, instruction); in PropagateLivenessToParameterCallers()
314 return FindOrDie(live_index_map_, instruction).element(shape_index); in IsLive()
Dhlo_clone_context.h68 return FindOrDie(instructions_, old_instruction); in GetInstruction()
73 return FindOrDie(computations_, old_computation); in GetComputation()
Dindexed_array_analysis.cc87 return FindOrDie(cache_, instr); in GetArrayFor()
110 switch (FindOrDie(dfs_state_map, instr)) { in TraverseAndPopulateCache()
142 instr->opcode(), FindOrDie(cache_, instr->operand(0)))); in ComputeArrayFor()
147 instr->opcode(), FindOrDie(cache_, instr->operand(0)), in ComputeArrayFor()
148 FindOrDie(cache_, instr->operand(1)))); in ComputeArrayFor()
157 FindOrDie(cache_, instr->operand(0)), in ComputeArrayFor()
158 FindOrDie(cache_, instr->operand(1)))); in ComputeArrayFor()
163 FindOrDie(cache_, instr->operand(0)))); in ComputeArrayFor()
169 FindOrDie(cache_, instr->operand(0)), in ComputeArrayFor()
170 FindOrDie(cache_, instr->operand(1)))); in ComputeArrayFor()
Dwhile_loop_invariant_code_motion.cc67 : FindOrDie(*hoisted_instructions, old_operand); in CreateLoopInvariantCopy()
272 FindOrDie(hoisted_instructions, instruction)); in TryHoistingInvariantInstructionsFromWhileBody()
288 FindOrDie(live_in_instructions_result.while_body_instruction_map, in TryHoistingInvariantInstructionsFromWhileBody()
Dlayout_assignment.cc537 FindOrDie(computation_layouts_, instruction->to_apply()); in AddMandatoryConstraints()
554 ComputationLayout& body_layout = FindOrDie(computation_layouts_, body); in AddMandatoryConstraints()
556 FindOrDie(computation_layouts_, condition); in AddMandatoryConstraints()
605 FindOrDie(computation_layouts_, in AddMandatoryConstraints()
612 FindOrDie(computation_layouts_, instruction->branch_computation(j)); in AddMandatoryConstraints()
918 FindOrDie(computation_layouts_, instruction->to_apply()))); in CheckLayouts()
929 FindOrDie(computation_layouts_, instruction->parent()))); in CheckLayouts()
937 FindOrDie(computation_layouts_, instruction->while_condition()), in CheckLayouts()
938 FindOrDie(computation_layouts_, instruction->while_body()))); in CheckLayouts()
944 FindOrDie(computation_layouts_, branch_computation)); in CheckLayouts()
[all …]
Dinstruction_fusion.cc415 FindOrDie(post_order_index_, instruction->mutable_operand(i)) > in DequeueNextInstructionAndOperandsToFuseInOrder()
416 FindOrDie(post_order_index_, instruction->mutable_operand(j))); in DequeueNextInstructionAndOperandsToFuseInOrder()
432 post_order_[FindOrDie(post_order_index_, instruction)] = nullptr; in RemoveInstruction()
Dmulti_output_fusion.h151 return FindOrDie(candidates_index_, instr); in get_candidate_id()
Dhlo_reachability.h176 return FindOrDie(indices_, GetKey(instruction)); in GetIndex()
Dhlo_domain_map.cc58 return FindOrDie(domain_metadata_id_, instruction); in GetDomainMetadataId()
Dheap_simulator.cc482 Chunk chunk = FindOrDie(result.chunk_map, group->canonical); in Finish()
757 BufferInterval& buffer_interval = FindOrDie(buffer_intervals_, buffer); in Free()
Dhlo_module.cc471 builder.Build(FindOrDie(outlined_instructions, output))); in OutlineExpressionFromComputation()
Dbuffer_assignment.cc229 const OffsetSize os = FindOrDie(assigned_buffers_, &buffer); in GetSlice()
317 const OffsetSize& offset_size = FindOrDie(assigned_buffers_, buffer); in ToString()
Dhlo_instructions.cc1302 FindOrDie(old_to_new, unfused_instruction->mutable_operand(index)); in MergeFusionInstructionIntoMultiOutput()
/external/tensorflow/tensorflow/compiler/xla/
Dmap_util.h34 const typename Collection::value_type::second_type& FindOrDie( in FindOrDie() function
44 typename Collection::value_type::second_type& FindOrDie( in FindOrDie() function
/external/protobuf/src/google/protobuf/stubs/
Dmap_util.h74 FindOrDie(const Collection& collection,
84 FindOrDie(Collection& collection, // NOLINT
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc576 auto less_than_function = FindOrDie(emitted_functions_, sort->to_apply()); in HandleSort()
1389 if (FindOrDie(unreduced_dim_map, operand_dim) != in ReductionPreservesLayout()
2204 llvm::Function* call_ir_function = FindOrDie(emitted_functions_, computation); in HandleCall()
2925 FindOrDie(constant_buffer_to_global_, slice.allocation()->index()), in EmitBufferPointer()
3067 Call(FindOrDie(emitted_functions_, &callee), in EmitThreadLocalCall()
3083 Call(FindOrDie(emitted_functions_, &callee), in EmitGlobalCall()
Dcpu_compiler.cc208 {hlo_instruction, FindOrDie(assigned_indices_, hlo_instruction)}); in DefaultAction()
/external/protobuf/src/google/protobuf/compiler/
Dparser_unittest.cc2232 pair<int, int> start_pos = FindOrDie(markers_, start_marker); in HasSpanWithComment()
2233 pair<int, int> end_pos = FindOrDie(markers_, end_marker); in HasSpanWithComment()