| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_clone_context.h | 46 instructions_[old_instruction] = new_instruction; in MapInstruction()
57 return FindOrDefault(instructions_, old_instruction, nullptr); in FindInstruction()
68 return FindOrDie(instructions_, old_instruction); in GetInstruction()
78 return instructions_; in cloned_instructions()
92 absl::flat_hash_map<const HloInstruction*, HloInstruction*> instructions_; variable
|
| D | hlo_computation.h | 94 instructions_.push_back(std::move(instruction)); in AddInstruction()
95 last_added_instruction_ = instructions_.back().get(); in AddInstruction()
101 for (const auto& instruction : instructions_) { in ForEachInstruction()
115 std::vector<std::unique_ptr<HloInstruction>> instructions_; variable
306 return {MakeUnwrappingIterator(instructions_.begin()), in instructions()
307 MakeUnwrappingIterator(instructions_.end())}; in instructions()
310 return {MakeUnwrappingIterator(instructions_.begin()), in instructions()
311 MakeUnwrappingIterator(instructions_.end())}; in instructions()
746 InstructionList instructions_; variable
|
| D | hlo_computation.cc | 62 for (auto& instruction : instructions_) { in Build()
72 name_, parameter_count, &instructions_, root, fusion_instruction_)); in Build()
149 instructions_.insert(instructions_.end(), std::move(instruction)); in AddInstructionInternal()
161 return instructions_.back().get(); in AddParameter()
179 return instructions_.back().get(); in AddEntryComputationParameter()
372 instructions_.erase(inst_it->second); in RemoveInstructionImpl()
388 for (auto& instruction : instructions_) { in set_root_instruction()
505 for (const auto& instruction : instructions_) { in ComputeChannelDependencies()
525 for (auto& instruction : instructions_) { in MakeInstructionPostOrder()
533 CHECK_EQ(instructions_.size(), post_order.size()) in MakeInstructionPostOrder()
[all …]
|
| D | buffer_assignment_test.cc | 64 instructions_.push_back(hlo); in DefaultAction()
69 std::vector<const HloInstruction*> GetInstructions() { return instructions_; } in GetInstructions()
76 std::vector<const HloInstruction*> instructions_; member in xla::__anon8b27d6dd0111::InstructionListVisitor
|
| /external/swiftshader/third_party/SPIRV-Tools/tools/sva/src/ |
| D | ast.js | 17 this.instructions_ = []; property
35 instructions() { return this.instructions_; }
37 instruction(val) { return this.instructions_[val]; }
40 this.instructions_.push(inst);
|
| /external/deqp-deps/SPIRV-Tools/tools/sva/src/ |
| D | ast.js | 17 this.instructions_ = []; property
35 instructions() { return this.instructions_; }
37 instruction(val) { return this.instructions_[val]; }
40 this.instructions_.push(inst);
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/tools/sva/src/ |
| D | ast.js | 17 this.instructions_ = []; property
35 instructions() { return this.instructions_; }
37 instruction(val) { return this.instructions_[val]; }
40 this.instructions_.push(inst);
|
| /external/cronet/third_party/boringssl/src/crypto/test/ |
| D | file_test.cc | 190 instructions_[key] = value; in ReadNext()
223 for (const auto &kv : instructions_) { in ReadNext()
277 return instructions_.count(key) > 0; in HasInstruction()
282 auto iter = instructions_.find(key); in GetInstruction()
283 if (iter == instructions_.end()) { in GetInstruction()
299 return instructions_[key]; in GetInstructionOrDie()
329 instructions_.clear(); in ClearInstructions()
361 instructions_[key] = value; in InjectInstruction()
|
| D | file_test.h | 230 std::map<std::string, std::string> instructions_; variable
|
| /external/boringssl/src/crypto/test/ |
| D | file_test.cc | 190 instructions_[key] = value; in ReadNext()
223 for (const auto &kv : instructions_) { in ReadNext()
277 return instructions_.count(key) > 0; in HasInstruction()
282 auto iter = instructions_.find(key); in GetInstruction()
283 if (iter == instructions_.end()) { in GetInstruction()
299 return instructions_[key]; in GetInstructionOrDie()
329 instructions_.clear(); in ClearInstructions()
361 instructions_[key] = value; in InjectInstruction()
|
| D | file_test.h | 230 std::map<std::string, std::string> instructions_; variable
|
| /external/rust/crates/quiche/deps/boringssl/src/crypto/test/ |
| D | file_test.cc | 188 instructions_[key] = value; in ReadNext()
221 for (const auto &kv : instructions_) { in ReadNext()
275 return instructions_.count(key) > 0; in HasInstruction()
280 auto iter = instructions_.find(key); in GetInstruction()
281 if (iter == instructions_.end()) { in GetInstruction()
297 return instructions_[key]; in GetInstructionOrDie()
327 instructions_.clear(); in ClearInstructions()
359 instructions_[key] = value; in InjectInstruction()
|
| D | file_test.h | 230 std::map<std::string, std::string> instructions_; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.h | 89 return instructions_.at(hlo); in derived_instructions()
94 instructions_[hlo]; in set_visiting_hlo()
115 HloInstructionMap<std::vector<HloInstruction*>> instructions_; variable
|
| D | spmd_partitioner.cc | 249 instructions_[visiting_hlo_].push_back(hlo); in AddInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 176 ->instructions_[op.builder()->handle_to_index_[op.handle_]]; in GetInstruction()
181 return &builder->instructions_[builder->handle_to_index_[handle]]; in GetInstructionByHandle()
329 for (const HloInstructionProto& instr : instructions_) { in GetProgramShape()
345 TF_RET_CHECK(!instructions_.empty()); in GetProgramShape()
346 return GetProgramShape(instructions_.back().id()); in GetProgramShape()
445 for (size_t index = 0; index < instructions_.size(); ++index) { in SetDynamicBinding()
446 HloInstructionProto& instr = instructions_[index]; in SetDynamicBinding()
508 return Build(instructions_.back().id(), remove_dynamic_dimensions); in Build()
537 for (size_t index = 0; index < instructions_.size(); ++index) { in Build()
539 *instructions_[index].mutable_shape() = in Build()
[all …]
|
| D | xla_builder.h | 1045 std::deque<HloInstructionProto> instructions_; variable
1551 return const_cast<InstructionType>(&instructions_.at(it->second)); in LookUpInstructionByHandleInternal()
|