Searched refs:getBroadcastDimensionsAttr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_utils.td | 45 "hlo::getBroadcastDimensionsAttr(&$_builder, $0, $1)">;
48 "hlo::getBroadcastDimensionsAttr(&$_builder, $0, $1, /*allow_empty=*/false)">;
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/utils/ |
| D | hlo_utils.h | 32 mlir::DenseIntElementsAttr getBroadcastDimensionsAttr(mlir::Builder* b,
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/ |
| D | hlo_utils.cc | 25 DenseIntElementsAttr getBroadcastDimensionsAttr(Builder *b, Value x, Value y, in getBroadcastDimensionsAttr() function
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3443 hlo::getBroadcastDimensionsAttr(&rewriter, iota, op.delta())); in matchAndRewrite()
3446 hlo::getBroadcastDimensionsAttr(&rewriter, scaled, op.start())); in matchAndRewrite()
3523 hlo::getBroadcastDimensionsAttr(&rewriter, iota, delta_cast)); in matchAndRewrite()
3526 hlo::getBroadcastDimensionsAttr(&rewriter, scaled, start_out_cast)); in matchAndRewrite()
3574 hlo::getBroadcastDimensionsAttr(&rewriter, op.stop(), op.start())); in matchAndRewrite()
3582 hlo::getBroadcastDimensionsAttr(&rewriter, step_denominator, one)); in matchAndRewrite()
3586 hlo::getBroadcastDimensionsAttr(&rewriter, step_numerator, in matchAndRewrite()
3594 hlo::getBroadcastDimensionsAttr(&rewriter, iota, step)); in matchAndRewrite()
3597 hlo::getBroadcastDimensionsAttr(&rewriter, scaled, op.start())); in matchAndRewrite()
|