Home
last modified time | relevance | path

Searched defs:embedding_attr (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_update_embedding_enqueue_op_inputs.cc53 auto embedding_attr = op->getAttrOfType<StringAttr>(kTPUEmbeddingAttr); in ExtractEmbeddingAttribute() local
100 const auto& embedding_attr = it.getKey(); in UpdateEmbeddingEnqueueOpInput() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc1038 const AttrValue* embedding_attr = in GetTPUEmbeddingInputNodes() local