Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/
Dlhlo_elemental_utils.cc137 auto inputDim = b->create<arith::AddIOp>( in elementalLower() local
168 int inputDim = std::distance(broadcastDimensions.begin(), it); in elementalLowerImplForBroadcastInDimOps() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/thlo/IR/
Dthlo_ops.cc601 int64_t inputDim = inputDims[permutationRef[i]]; in verify() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DStablehloOps.cpp4724 static int64_t inferSliceDim(int64_t inputDim, int64_t start, int64_t end, in inferSliceDim()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc6796 static int64_t inferSliceDim(int64_t inputDim, int64_t start, int64_t end, in inferSliceDim()