Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dunfuse_batch_norm.cc37 Value value_1d, Value shape_value, in BroadcastToFeatureDim() argument
45 loc, result_type, value_1d, shape_value, dims); in BroadcastToFeatureDim()
48 return rewriter.create<mhlo::BroadcastInDimOp>(loc, result_type, value_1d, in BroadcastToFeatureDim()