Searched refs:GetReshapeOutputType (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 577 LogicalResult GetReshapeOutputType(Value tensor, Value shape, in GetReshapeOutputType() function
673 if (failed(GetReshapeOutputType(op.tensor(), op.shape(), error_handler, in Verify()
706 if (failed(GetReshapeOutputType(tensor, shape, error_handler, output_ty))) in build()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1349 LogicalResult GetReshapeOutputType(Value input, Value shape, in GetReshapeOutputType() function
1447 if (failed(GetReshapeOutputType(op.input(), op.shape(), error_handler, in Verify()
|