Searched refs:ResourceUpdate (Results 1 – 11 of 11) sorted by relevance
27 // CHECK-NEXT: // ResourceUpdate input_index=1 type=float shape=(2) modified
66 # CHECK-NEXT: // ResourceUpdate input_index=1 type=float shape=(2) modified
243 [&](const XlaCompiler::ResourceUpdate& update) { in PopulateInputs()388 const XlaCompiler::ResourceUpdate& write) { in GetOrCreateResourceVar()405 const XlaCompiler::ResourceUpdate& write = in GatherVariableInfo()538 const XlaCompiler::ResourceUpdate& write = in PopulateOutputs()
115 for (const XlaCompiler::ResourceUpdate& update : result.resource_updates) { in ConvertGraphToXla()
137 using ResourceUpdate = ::tensorflow::XlaResourceUpdate; variable
177 std::vector<XlaCompiler::ResourceUpdate>* resource_updates, in BuildComputation()300 XlaCompiler::ResourceUpdate& update = resource_updates->back(); in BuildComputation()
832 const XlaCompiler::ResourceUpdate& update = result.resource_updates[0]; in TEST_F()
171 for (const XlaCompiler::ResourceUpdate& update : result.resource_updates) { in Compile()351 const XlaCompiler::ResourceUpdate& update = result.resource_updates[i]; in Compile()
59 for (const XlaCompiler::ResourceUpdate& update : result->resource_updates) { in PopulateTensorArrayGradients()336 const XlaCompiler::ResourceUpdate& update = result->resource_updates[i]; in Compile()
370 const XlaCompiler::ResourceUpdate& update = body.resource_updates[i]; in Compile()607 const XlaCompiler::ResourceUpdate& update = body.resource_updates[i]; in Compile()
215 const XlaCompiler::ResourceUpdate& update = in AddVariableUpdatesToCores()