Home
last modified time | relevance | path

Searched refs:GatherInternal (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.h189 StatusOr<XlaOp> GatherInternal(
Dmlir_hlo_builder.cc381 StatusOr<XlaOp> MlirHloBuilder::GatherInternal( in GatherInternal() function in xla::MlirHloBuilder
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2406 return GatherInternal(shape, input, start_indices, dimension_numbers, in Gather()
2411 StatusOr<XlaOp> XlaBuilder::GatherInternal( in GatherInternal() function in xla::XlaBuilder
Dxla_builder.h878 virtual StatusOr<XlaOp> GatherInternal(