Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DStablehloOps.cpp6081 int64_t outBatchDimension = 0; in parseConvolutionDimensionsRaw() local
6290 const int64_t outBatchDimension = parsedDims.second[IOBatch]; in parseConvolutionDimensions() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc8907 int64_t outBatchDimension = 0; in parseConvolutionDimensionsRaw() local
9116 const int64_t outBatchDimension = parsedDims.second[IOBatch]; in parseConvolutionDimensions() local