Home
last modified time | relevance | path

Searched defs:getStartIndicesDim (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DStablehloOps.cpp1492 int64_t resultRank, llvm::function_ref<dimTy(int64_t)> getStartIndicesDim, in inferGatherShape()
1582 auto getStartIndicesDim = [&](int64_t index) { in inferGatherReturnTypeComponents() local
1617 auto getStartIndicesDim = [&](int64_t index) { in reifyGatherShape() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc1604 int64_t resultRank, llvm::function_ref<dimTy(int64_t)> getStartIndicesDim, in inferGatherShape()
1694 auto getStartIndicesDim = [&](int64_t index) { in inferGatherReturnTypeComponents() local
1729 auto getStartIndicesDim = [&](int64_t index) { in reifyGatherShape() local