Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_dimension_inference_test.cc1275 auto dynamic_reshape = in TEST_F() local
1302 auto dynamic_reshape = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F() local
H A Ddfs_hlo_visitor_with_default.h214 Status HandleDynamicReshape(HloInstructionPtr dynamic_reshape) override { in HandleDynamicReshape()
H A Dhlo_verifier.cc1085 Status ShapeVerifier::HandleDynamicReshape(HloInstruction* dynamic_reshape) { in HandleDynamicReshape()
H A Ddynamic_dimension_inference.cc915 HloDynamicReshapeInstruction* dynamic_reshape = in HandleDynamicReshape() local