Home
last modified time | relevance | path

Searched defs:reshape1 (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dreshape_mover_test.cc50 auto reshape1 = in TEST_F() local
119 auto reshape1 = in TEST_F() local
146 auto reshape1 = in TEST_F() local
194 auto reshape1 = variable
277 auto reshape1 = variable
369 auto reshape1 = variable
398 auto reshape1 = in TEST_F() local
431 auto reshape1 = in TEST_F() local
506 auto reshape1 = in TEST_F() local
H A Dpattern_matcher_test.cc997 auto reshape1 = in TEST_F() local
H A Dalgebraic_simplifier_test.cc2283 auto reshape1 = builder.AddInstruction( in TEST_F() local
3120 HloInstruction* reshape1 = in TEST_F() local
3313 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dcpu_gpu_fusion_test.cc348 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F() local
363 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local
378 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local
393 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F() local
408 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local
423 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local
438 auto reshape1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F() local
453 auto reshape1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F() local
560 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local
H A Dvalue_inference_test.cc190 auto reshape1 = Reshape(slice1, {}); in TEST_F() local
/aosp_15_r20/external/armnn/src/armnn/test/optimizations/
H A DOptimizeConsecutiveReshapesTests.cpp35 auto reshape1 = graph.InsertNewLayer<armnn::ReshapeLayer>( variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_instruction_fusion_test.cc414 HloInstruction* reshape1 = builder.AddInstruction( in TEST_F() local