Home
last modified time | relevance | path

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

/external/armnn/src/armnn/test/optimizations/
DOptimizeConsecutiveReshapesTests.cpp36 auto reshape2 = graph.InsertNewLayer<armnn::ReshapeLayer>( in BOOST_AUTO_TEST_CASE() local
/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.cc5368 auto reshape2 = in TEST_F() local