Searched refs:resource_op (Results 1 – 4 of 4) sorted by relevance
146 string ResourceOpToString(const ResourceOp& resource_op) { in ResourceOpToString() argument148 resource_op.first, ": ", in ResourceOpToString()149 XlaResourceOpInfo::XlaResourceOpKindToString(resource_op.second)); in ResourceOpToString()178 for (ResourceOp resource_op : other) { in Add() local179 Add(resource_op); in Add()183 void Add(const ResourceOp& resource_op) { in Add() argument185 if (!IsCopy() && Contains(resource_op)) { in Add()191 impl_->insert(resource_op); in Add()202 bool Contains(const ResourceOp& resource_op) const { in Contains()203 return impl_ != nullptr && impl_->count(resource_op); in Contains()
107 std::unique_ptr<StubResourceOpKernel> resource_op( in CreateOp() local109 EXPECT_TRUE(resource_op != nullptr); in CreateOp()110 if (resource_op != nullptr) { in CreateOp()113 return resource_op; in CreateOp()
79 auto resource_op = in InitializeVariable() local84 resource_op, value_op); in InitializeVariable()
158 if (auto* resource_op = resource.getDefiningOp()) { in BuildVariableAccessInfo() local159 auto resource_attr = resource_op->getAttr(kDeviceAttr); in BuildVariableAccessInfo()