Home
last modified time | relevance | path

Searched defs:reshape2 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover_test.cc198 auto reshape2 = variable
509 auto reshape2 = in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion_test.cc81 HloInstruction* reshape2 = builder.AddInstruction( in TEST_F() local
118 auto reshape2 = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc321 HloInstruction* reshape2 = in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc4992 auto reshape2 = in TEST_F() local