Home
last modified time | relevance | path

Searched refs:new_producer (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction.cc1890 HloInstruction* new_producer) { in ReplaceUseWith() argument
1892 ShapeUtil::CompatibleIgnoringFpPrecision(shape(), new_producer->shape())) in ReplaceUseWith()
1895 << ShapeUtil::HumanString(new_producer->shape()); in ReplaceUseWith()
1896 return ReplaceUseWithDifferentShape(user, new_producer); in ReplaceUseWith()
1900 HloInstruction* user, HloInstruction* new_producer) { in ReplaceUseWithDifferentShape() argument
1902 << " with " << new_producer->name(); in ReplaceUseWithDifferentShape()
1908 new_producer); in ReplaceUseWithDifferentShape()
1909 new_producer->AddUser(user); in ReplaceUseWithDifferentShape()
1948 Status HloInstruction::ReplaceAllUsesWith(HloInstruction* new_producer) { in ReplaceAllUsesWith() argument
1950 ShapeUtil::CompatibleIgnoringFpPrecision(shape(), new_producer->shape())) in ReplaceAllUsesWith()
[all …]
Dhlo_instruction.h964 Status ReplaceUseWith(HloInstruction* user, HloInstruction* new_producer);
968 HloInstruction* new_producer);
993 Status ReplaceAllUsesWith(HloInstruction* new_producer);
996 Status ReplaceAllUsesWithDifferentShape(HloInstruction* new_producer);
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc1976 NodeDef* new_producer = AddCopyNode(optimized_consumer_name, consumer); in TrySimplify() local
1977 new_producer->set_input(0, producer->input(0)); in TrySimplify()
1978 ctx().node_map->AddOutput(input->name(), new_producer->name()); in TrySimplify()
1982 new_consumer->set_input(0, new_producer->name()); in TrySimplify()
1985 producer_is_cast ? new_producer : new_consumer; in TrySimplify()
1994 ctx().node_map->AddOutput(new_producer->name(), new_consumer->name()); in TrySimplify()
1996 AddToOptimizationQueue(new_producer); in TrySimplify()