Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dresampler_ops.cc140 xla::GatherDimensionNumbers gather_dim_numbers; in Gather2by2Neighbors() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc439 GatherDimensionNumbers gather_dim_numbers; in RewriteDynamicReshapeSplitInput() local
660 GatherDimensionNumbers gather_dim_numbers; in RewriteDynamicReshapeCombineInput() local
Dshape_inference.cc3441 const GatherDimensionNumbers& gather_dim_numbers, in InferGatherShape()
Dhlo_instructions.cc2889 const GatherDimensionNumbers& gather_dim_numbers, in HloGatherInstruction()
2923 GatherDimensionNumbers gather_dim_numbers; in MakeGatherDimNumbers() local
Dhlo_instruction.cc1804 const GatherDimensionNumbers& gather_dim_numbers, in CreateGather()