Home
last modified time | relevance | path

Searched defs:gather_dim_numbers (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dunique_op.cc194 xla::GatherDimensionNumbers gather_dim_numbers; in CompileWithAxis() local
H A Dresampler_ops.cc143 xla::GatherDimensionNumbers gather_dim_numbers; in Gather2by2Neighbors() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_padder.cc530 GatherDimensionNumbers gather_dim_numbers; in RewriteDynamicReshapeSplitInput() local
715 GatherDimensionNumbers gather_dim_numbers; in RewriteDynamicReshapeCombineInput() local
H A Dhlo_instructions.cc3049 const GatherDimensionNumbers& gather_dim_numbers, in HloGatherInstruction()
3083 GatherDimensionNumbers gather_dim_numbers; in MakeGatherDimNumbers() local
H A Dshape_inference.cc3423 const GatherDimensionNumbers& gather_dim_numbers, in InferGatherShape()
H A Dhlo_instruction.cc1928 const GatherDimensionNumbers& gather_dim_numbers, in CreateGather()