Home
last modified time | relevance | path

Searched refs:tensor_array_gradients_accessed (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcase_op.cc181 for (const string& grad_source : update.tensor_array_gradients_accessed) { in Compile()
265 lhs.tensor_array_gradients_accessed == in Compile()
266 rhs.tensor_array_gradients_accessed; in Compile()
Dif_op.cc70 for (const string& grad_source : update.tensor_array_gradients_accessed) { in PopulateTensorArrayGradients()
175 lhs.tensor_array_gradients_accessed == in ValidateShapes()
176 rhs.tensor_array_gradients_accessed; in ValidateShapes()
Dwhile_op.cc401 for (const string& grad_source : update.tensor_array_gradients_accessed) { in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.h127 std::set<string> tensor_array_gradients_accessed; member
Dxla_compiler.cc325 update.tensor_array_gradients_accessed.insert(grad.first); in BuildComputation()
Dxla_compiler_test.cc842 update.tensor_array_gradients_accessed); in TEST_F()