Searched refs:kTFImplements (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | perception_ops_utils.cc | 32 constexpr char kTFImplements[] = "tf._implements"; variable
73 func_->setAttr(kTFImplements, in RewriteFunc()
150 func_->setAttr(kTFImplements, in RewriteFunc()
|
| D | lstm_utils_test.cc | 84 func->setAttr(kTFImplements, attr_values); in createLstmCompositeFunc()
129 fused_lstm_func_->getAttrOfType<StringAttr>(kTFImplements).getValue(), in TEST_F()
202 EXPECT_EQ(fused_lstm_func_cifg_->getAttrOfType<StringAttr>(kTFImplements) in TEST_F()
227 fused_ln_lstm_func_->getAttrOfType<StringAttr>(kTFImplements).getValue(), in TEST_F()
|
| D | nms_utils.cc | 27 constexpr char kTFImplements[] = "tf._implements"; variable
43 func_->setAttr(kTFImplements, in RewriteFunc()
88 func_->setAttr(kTFImplements, in RewriteFunc()
|
| D | tftext_utils.cc | 51 constexpr char kTFImplements[] = "tf._implements"; variable
136 func->setAttr(kTFImplements, attr); in ConvertWhitespaceTokenizer()
259 func->setAttr(kTFImplements, attr); in ConvertNgrams()
339 func->setAttr(kTFImplements, attr); in ConvertSgnnProjection()
|
| D | lstm_utils.h | 34 constexpr char kTFImplements[] = "tf._implements"; variable
|
| D | lstm_utils.cc | 442 auto attr = fused_func_op_->getAttrOfType<StringAttr>(kTFImplements); in InitializeFromFuncAttributes()
445 << "Invalid function attribute, expected " << kTFImplements in InitializeFromFuncAttributes()
456 << kTFImplements << " attribute should be set"; in InitializeFromFuncAttributes()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_composite_functions_tf.cc | 76 func_->setAttr(kTFImplements, in RewriteFunc()
345 func->getAttrOfType<StringAttr>(kTFImplements); in runOnOperation()
351 auto tf_implements_attr = func->getAttrOfType<FuncAttr>(kTFImplements); in runOnOperation()
|