Searched refs:resource_updates (Results 1 – 7 of 7) sorted by relevance
114 for (const XlaCompiler::ResourceUpdate& update : result->resource_updates) { in Compile()184 then_result.resource_updates.size() == in Compile()185 else_result.resource_updates.size(), in Compile()188 for (int i = 0; i < then_result.resource_updates.size(); ++i) { in Compile()189 const auto& lhs = then_result.resource_updates[i]; in Compile()190 const auto& rhs = else_result.resource_updates[i]; in Compile()261 for (int i = 0; i < result->resource_updates.size(); ++i) { in Compile()262 const XlaCompiler::ResourceUpdate& update = result->resource_updates[i]; in Compile()
113 for (const XlaCompiler::ResourceUpdate& update : result->resource_updates) { in Compile()193 branch_results[0].resource_updates.size() == in Compile()194 branch_results[j].resource_updates.size(), in Compile()197 for (int i = 0; i < branch_results[0].resource_updates.size(); ++i) { in Compile()198 const auto& lhs = branch_results[0].resource_updates[i]; in Compile()199 const auto& rhs = branch_results[j].resource_updates[i]; in Compile()273 for (int i = 0; i < result.resource_updates.size(); ++i) { in Compile()274 const XlaCompiler::ResourceUpdate& update = result.resource_updates[i]; in Compile()
348 for (int i = 0; i < body.resource_updates.size(); ++i) { in Compile()349 const XlaCompiler::ResourceUpdate& update = body.resource_updates[i]; in Compile()553 for (int i = 0; i < body.resource_updates.size(); ++i) { in Compile()554 const XlaCompiler::ResourceUpdate& update = body.resource_updates[i]; in Compile()
387 variable_infos.reserve(kernel->resource_updates.size()); in PopulateOutputs()389 for (int i = 0; i < kernel->resource_updates.size(); ++i) { in PopulateOutputs()390 const XlaCompiler::ResourceUpdate& write = kernel->resource_updates[i]; in PopulateOutputs()410 for (int i = 0; i < kernel->resource_updates.size(); ++i) { in PopulateOutputs()412 const XlaCompiler::ResourceUpdate& write = kernel->resource_updates[i]; in PopulateOutputs()
274 std::vector<ResourceUpdate> resource_updates; member
168 std::vector<XlaCompiler::ResourceUpdate>* resource_updates, in BuildComputation() argument272 resource_updates->emplace_back(); in BuildComputation()273 XlaCompiler::ResourceUpdate& update = resource_updates->back(); in BuildComputation()1169 &result->resource_updates, &result->xla_output_shape)); in CompileGraph()
763 ASSERT_EQ(1, result.resource_updates.size()); in TEST_F()764 const XlaCompiler::ResourceUpdate& update = result.resource_updates[0]; in TEST_F()823 EXPECT_EQ(0, result.resource_updates.size()); in TEST_F()856 EXPECT_EQ(1, result.resource_updates.size()); in TEST_F()