Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dresampler_ops.cc173 xla::ScatterDimensionNumbers scatter_dim_numbers; in ScatterToGradData() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc3660 const ScatterDimensionNumbers& scatter_dim_numbers) { in InferScatterShape()
Dhlo_instruction.cc1939 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in CreateScatter()
1950 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in CreateScatter()
Dhlo_instructions.cc3143 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in HloScatterInstruction()
3184 ScatterDimensionNumbers scatter_dim_numbers; in MakeScatterDimNumbers() local