Home
last modified time | relevance | path

Searched refs:attr_slice (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dcwise_op_vectorizer.cc125 for (const auto& attr_slice : node.attrs()) { in CwiseVectorizeHelper() local
126 node_builder = node_builder.Attr(attr_slice.first, attr_slice.second); in CwiseVectorizeHelper()
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util_test.cc471 const AttrSlice attr_slice = in TEST() local
476 EXPECT_TRUE(OutputTypesForNode(attr_slice, op_def, &types).ok()); in TEST()
Dfunction.cc695 AttrSlice attr_slice(attrs); in AddDefaultAttrs() local
697 if (attr_def.has_default_value() && !attr_slice.Find(attr_def.name())) { in AddDefaultAttrs()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc809 AttrSlice attr_slice = AttrSlice(&remote_op.attrs()); in StoreResourceDtypesAndShapes() local
811 TF_RETURN_IF_ERROR(attr_slice.Find("dtype", &dtype)); in StoreResourceDtypesAndShapes()
813 TF_RETURN_IF_ERROR(attr_slice.Find("shape", &shape)); in StoreResourceDtypesAndShapes()