Home
last modified time | relevance | path

Searched refs:resource_updates (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dif_op.cc59 for (const XlaCompiler::ResourceUpdate& update : result->resource_updates) { in PopulateTensorArrayGradients()
146 if (then_result.resource_updates.size() != in ValidateShapes()
147 else_result.resource_updates.size()) { in ValidateShapes()
152 for (int i = 0; i < then_result.resource_updates.size(); ++i) { in ValidateShapes()
153 const auto& lhs = then_result.resource_updates[i]; in ValidateShapes()
154 const auto& rhs = else_result.resource_updates[i]; in ValidateShapes()
335 for (int i = 0; i < result->resource_updates.size(); ++i) { in Compile()
336 const XlaCompiler::ResourceUpdate& update = result->resource_updates[i]; in Compile()
Dcase_op.cc171 for (const XlaCompiler::ResourceUpdate& update : result.resource_updates) { in Compile()
265 branch_results[0].resource_updates.size() == in Compile()
266 branch_results[j].resource_updates.size(), in Compile()
269 for (int i = 0; i < branch_results[0].resource_updates.size(); ++i) { in Compile()
270 const auto& lhs = branch_results[0].resource_updates[i]; in Compile()
271 const auto& rhs = branch_results[j].resource_updates[i]; in Compile()
350 for (int i = 0; i < result.resource_updates.size(); ++i) { in Compile()
351 const XlaCompiler::ResourceUpdate& update = result.resource_updates[i]; in Compile()
Dwhile_op.cc369 for (int i = 0; i < body.resource_updates.size(); ++i) { in Compile()
370 const XlaCompiler::ResourceUpdate& update = body.resource_updates[i]; in Compile()
606 for (int i = 0; i < body.resource_updates.size(); ++i) { in Compile()
607 const XlaCompiler::ResourceUpdate& update = body.resource_updates[i]; in Compile()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc242 absl::c_any_of(compilation_result->resource_updates, in PopulateInputs()
403 out.reserve(compilation_result.resource_updates.size()); in GatherVariableInfo()
404 for (int i = 0; i < compilation_result.resource_updates.size(); ++i) { in GatherVariableInfo()
406 compilation_result.resource_updates[i]; in GatherVariableInfo()
536 for (int i = 0, end = compilation_result->resource_updates.size(); i < end; in PopulateOutputs()
539 compilation_result->resource_updates[i]; in PopulateOutputs()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc316 for (int64 i = 0, end = result.resource_updates.size(); i < end; i++) { in CompileFunctionalNode()
317 if (result.resource_updates[i].modified) { in CompileFunctionalNode()
319 expressions[result.resource_updates[i].input_index]->resource(); in CompileFunctionalNode()
Dxla_helpers.h164 std::vector<XlaResourceUpdate> resource_updates; member
Dtf2xla.cc115 for (const XlaCompiler::ResourceUpdate& update : result.resource_updates) { in ConvertGraphToXla()
Dxla_compiler.cc177 std::vector<XlaCompiler::ResourceUpdate>* resource_updates, in BuildComputation() argument
299 resource_updates->emplace_back(); in BuildComputation()
300 XlaCompiler::ResourceUpdate& update = resource_updates->back(); in BuildComputation()
1394 &result->resource_updates, &result->xla_output_shape, in CompileGraph()
Dxla_compiler_test.cc831 ASSERT_EQ(1, result.resource_updates.size()); in TEST_F()
832 const XlaCompiler::ResourceUpdate& update = result.resource_updates[0]; in TEST_F()
890 EXPECT_EQ(0, result.resource_updates.size()); in TEST_F()
922 EXPECT_EQ(1, result.resource_updates.size()); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc140 std::vector<XlaResourceUpdate>* resource_updates) { in GetOutputInfo() argument
151 resource_updates->reserve(func_type.getNumResults()); in GetOutputInfo()
173 resource_updates->emplace_back(); in GetOutputInfo()
174 XlaResourceUpdate& resource_update = resource_updates->back(); in GetOutputInfo()
464 &compilation_result->resource_updates); in PopulateResultIOInfo()
Dtf_xla_mlir_translate.cc106 for (const auto& resource_update : compilation_result.resource_updates) { in PrintHloModuleText()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc214 if (resource_update_pos < compilation_result.resource_updates.size()) { in AddVariableUpdatesToCores()
216 compilation_result.resource_updates[resource_update_pos]; in AddVariableUpdatesToCores()