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/
Dif_op.cc67 for (const string& grad_source : update.tensor_array_gradients_accessed) { in PopulateTensorArrayGradients()
156 lhs.tensor_array_gradients_accessed == in ValidateShapes()
157 rhs.tensor_array_gradients_accessed; in ValidateShapes()
Dcase_op.cc179 for (const string& grad_source : update.tensor_array_gradients_accessed) { in Compile()
274 lhs.tensor_array_gradients_accessed == in Compile()
275 rhs.tensor_array_gradients_accessed; in Compile()
Dwhile_op.cc389 for (const string& grad_source : update.tensor_array_gradients_accessed) { in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.h133 std::set<string> tensor_array_gradients_accessed; member
Dxla_compiler.cc321 update.tensor_array_gradients_accessed.insert(grad.first); in BuildComputation()
Dxla_compiler_test.cc836 update.tensor_array_gradients_accessed); in TEST_F()