Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dunfuse_batch_norm.cc40 Value value1d, Value shapeValue, int64_t featureDim, in broadcastToFeatureDim()
120 int64_t featureDim = bnOp.feature_index(); in matchAndRewrite() local