Searched refs:new_producer (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | instruction_fusion.cc | 143 HloInstruction* new_producer = in FuseInstruction() local
145 evaluation->second.UpdateEvaluationCache(new_producer, indexing_users); in FuseInstruction()
146 return new_producer; in FuseInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | fusion_node_indexing_evaluation_test.cc | 53 HloInstruction* new_producer = in FuseInstruction() local
55 evaluation->second.UpdateEvaluationCache(new_producer, indexing_users); in FuseInstruction()
56 return new_producer; in FuseInstruction()
|
| D | hlo_instruction.cc | 2301 HloInstruction* new_producer) { in ReplaceUseWith() argument
2303 ShapeUtil::CompatibleIgnoringFpPrecision(shape(), new_producer->shape())) in ReplaceUseWith()
2306 << ShapeUtil::HumanString(new_producer->shape()); in ReplaceUseWith()
2307 return ReplaceUseWithDifferentShape(user, new_producer); in ReplaceUseWith()
2311 HloInstruction* user, HloInstruction* new_producer) { in ReplaceUseWithDifferentShape() argument
2313 << " with " << new_producer->name(); in ReplaceUseWithDifferentShape()
2319 new_producer); in ReplaceUseWithDifferentShape()
2320 new_producer->AddUser(user); in ReplaceUseWithDifferentShape()
2361 HloInstruction* new_producer) { in ReplaceUsesWith() argument
2363 ShapeUtil::CompatibleIgnoringFpPrecision(shape(), new_producer->shape())) in ReplaceUsesWith()
[all …]
|
| D | space_to_batch_converter.cc | 1000 auto new_producer = old_to_new_instrs_[producer]; in PropagateOnBroadcast() local
1001 auto permute_dims = instr_to_dim_permute_map_[new_producer]; in PropagateOnBroadcast()
1019 new_producer->shape())) { in PropagateOnBroadcast()
1026 new_producer->shape().dimensions().begin(), in PropagateOnBroadcast()
1027 new_producer->shape().dimensions().end()); in PropagateOnBroadcast()
1044 MakeReshapeHlo(new_producer->shape().dimensions(), new_broadcast) in PropagateOnBroadcast()
1243 auto new_producer = old_to_new_instrs_[producer]; in Propagate() local
1246 previous_broadcast->shape(), new_producer->shape())) { in Propagate()
2116 HloInstruction* new_producer = old_to_new_instrs_[producer]; in PropagateOnConstant() local
2117 auto prod_transpose_dims = instr_to_dim_permute_map_[new_producer]; in PropagateOnConstant()
[all …]
|
| D | hlo_instruction.h | 1186 Status ReplaceUseWith(HloInstruction* user, HloInstruction* new_producer);
1190 HloInstruction* new_producer);
1215 Status ReplaceAllUsesWith(HloInstruction* new_producer);
1218 Status ReplaceAllUsesWithDifferentShape(HloInstruction* new_producer);
1222 HloInstruction* new_producer);
1224 absl::Span<HloInstruction* const> users, HloInstruction* new_producer);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion.cc | 215 HloInstruction* new_producer = in FuseInstruction() local
217 evaluation->second.UpdateEvaluationCache(new_producer, indexing_users); in FuseInstruction()
218 return new_producer; in FuseInstruction()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 2079 NodeDef* new_producer = AddCopyNode(optimized_consumer_name, consumer); in TrySimplify() local
2080 new_producer->set_input(0, producer->input(0)); in TrySimplify()
2081 ctx().node_map->AddOutput(input->name(), new_producer->name()); in TrySimplify()
2085 new_consumer->set_input(0, new_producer->name()); in TrySimplify()
2088 producer_is_cast ? new_producer : new_consumer; in TrySimplify()
2097 ctx().node_map->AddOutput(new_producer->name(), new_consumer->name()); in TrySimplify()
2099 AddToOptimizationQueue(new_producer); in TrySimplify()
|