Home
last modified time | relevance | path

Searched defs:dynamic_slice_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dconvolution_handler.cc354 Shape dynamic_slice_shape = lhs.hlo()->shape(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_instruction_fusion_test.cc322 Shape dynamic_slice_shape = ShapeUtil::MakeShape(F32, {4, 4}); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier.cc2294 const Shape& dynamic_slice_shape = a->shape(); in OptimizeDotOfGather() local