Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dchlo_ops.cc160 SmallVectorImpl<Value>& reifiedReturnShapes) { in ReifyBroadcastBinaryOpReturnTypeShapes() argument
185 reifiedReturnShapes.push_back(computed_shape); in ReifyBroadcastBinaryOpReturnTypeShapes()
208 OpBuilder& builder, SmallVectorImpl<Value>& reifiedReturnShapes) { in reifyReturnTypeShapes() argument
210 reifiedReturnShapes); in reifyReturnTypeShapes()
239 OpBuilder& builder, SmallVectorImpl<Value>& reifiedReturnShapes) { in reifyReturnTypeShapes() argument
241 reifiedReturnShapes); in reifyReturnTypeShapes()
297 OpBuilder& builder, SmallVectorImpl<Value>& reifiedReturnShapes) { \
299 reifiedReturnShapes); \
Dhlo_ops.cc1877 OpBuilder& builder, SmallVectorImpl<Value>& reifiedReturnShapes) { in reifyReturnTypeShapes() argument
1879 &reifiedReturnShapes); in reifyReturnTypeShapes()
2851 OpBuilder& builder, SmallVectorImpl<Value>& reifiedReturnShapes) { in reifyReturnTypeShapes() argument
2853 &reifiedReturnShapes); in reifyReturnTypeShapes()
3168 SmallVectorImpl<Value>* reifiedReturnShapes) { in deriveShapeFromFirstOperand() argument
3189 *reifiedReturnShapes = SmallVector<Value, 1>{ in deriveShapeFromFirstOperand()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops.h82 SmallVectorImpl<Value> *reifiedReturnShapes);
Dhlo_ops.td136 OpBuilder& builder, SmallVectorImpl<Value>& reifiedReturnShapes) {
138 &reifiedReturnShapes);
290 OpBuilder& builder, SmallVectorImpl<Value>& reifiedReturnShapes) {
292 &reifiedReturnShapes);
/external/llvm-project/mlir/include/mlir/Interfaces/
DInferTypeOpInterface.td112 "SmallVectorImpl<Value>&":$reifiedReturnShapes),