Searched defs:dense_attr (Results 1 – 6 of 6) sorted by relevance
56 const DenseAttr& dense_attr, HostContext* host) { in DecodeDenseAttrToTfTensor()243 auto dense_attr = attrs.GetAsserting<tfrt::DenseAttr>(entry.name); in FillAttrValueMapUsingScalar() local557 } else if (auto dense_attr = bef_attr.dyn_cast<tfrt::DenseAttr>()) { in SetUpScalarAttr() local596 void AddTensorToAttrList(tfrt::DenseAttr dense_attr, in AddTensorToAttrList()609 } else if (auto dense_attr = base.dyn_cast<tfrt::DenseAttr>()) { in SetUpListAttr() local
415 auto dense_attr = attr.dyn_cast<mlir::DenseElementsAttr>(); in ConvertToTensorProto() local
386 const auto dense_attr = attr.dyn_cast_or_null<DenseElementsAttr>(); in GetNumElementsOrOne() local412 const auto dense_attr = attr.dyn_cast_or_null<DenseIntElementsAttr>(); in IsOneHotIndexAttribute() local
339 const DenseAttr& dense_attr, HostContext* host, in DecodeDenseAttrToTensorInterface()458 DenseAttr dense_attr = attrs.GetAsserting<DenseAttr>(entry.name); in PrepareAttributes() local
318 AttrValue dense_attr; in AsGraphDefInternal() local
1697 DenseAttr dense_attr(buffer.data() + offset); in SetAttrTensor() local