Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dreshape_mover_test.cc199 auto reshape2 = variable
510 auto reshape2 = in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dinstruction_fusion_test.cc81 HloInstruction* reshape2 = builder.AddInstruction( in TEST_F() local
118 auto reshape2 = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F() local
/aosp_15_r20/external/armnn/src/armnn/test/optimizations/
H A DOptimizeConsecutiveReshapesTests.cpp37 auto reshape2 = graph.InsertNewLayer<armnn::ReshapeLayer>( variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_instruction_fusion_test.cc303 HloInstruction* reshape2 = in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_test.cc7372 auto reshape2 = in TEST_F() local