Home
last modified time | relevance | path

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

/external/armnn/src/armnn/test/optimizations/
DOptimizeConsecutiveReshapesTests.cpp37 auto reshape2 = graph.InsertNewLayer<armnn::ReshapeLayer>( variable
/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover_test.cc199 auto reshape2 = variable
510 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.cc303 HloInstruction* reshape2 = in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc7372 auto reshape2 = in TEST_F() local