Home
last modified time | relevance | path

Searched refs:tensor_array_gradients (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dif_op.cc77 for (const auto& gradient : resource->tensor_array_gradients()) { in Compile()
78 arg.tensor_array_gradients.insert(gradient.first); in Compile()
131 for (const auto& gradient : resource->tensor_array_gradients()) { in Compile()
132 arg.tensor_array_gradients.insert(gradient.first); in Compile()
134 if (!resource->tensor_array_gradients().empty()) in Compile()
270 arguments[update.input_index].tensor_array_gradients, in Compile()
Dcase_op.cc74 for (const auto& gradient : resource->tensor_array_gradients()) { in Compile()
75 arg.tensor_array_gradients.insert(gradient.first); in Compile()
130 for (const auto& gradient : resource->tensor_array_gradients()) { in Compile()
131 arg.tensor_array_gradients.insert(gradient.first); in Compile()
133 if (!resource->tensor_array_gradients().empty()) { in Compile()
282 arguments[update.input_index].tensor_array_gradients, in Compile()
Dwhile_op.cc75 for (const auto& gradient : resource->tensor_array_gradients()) { in MakeXlaCompilerArgumentsFromInputs()
76 arg.tensor_array_gradients.insert(gradient.first); in MakeXlaCompilerArgumentsFromInputs()
378 for (const auto& gradient : resource->tensor_array_gradients()) { in Compile()
379 arg.tensor_array_gradients.insert(gradient.first); in Compile()
561 arguments[update.input_index].tensor_array_gradients, in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_resource.h53 const std::set<string>& tensor_array_gradients,
159 const std::map<string, std::unique_ptr<XlaResource>>& tensor_array_gradients() in tensor_array_gradients() function
Dxla_compiler.cc265 for (const auto& grad : resource->tensor_array_gradients()) { in BuildComputation()
269 arg.tensor_array_gradients.count(grad.first) == 0; in BuildComputation()
278 for (const auto& grad : resource->tensor_array_gradients()) { in BuildComputation()
345 tensor_array_gradients) != in operator ==()
348 other.tensor_array_gradients)) { in operator ==()
393 if (!tensor_array_gradients.empty()) { in HumanString()
395 absl::StrJoin(tensor_array_gradients, ",")); in HumanString()
662 if (!arg.tensor_array_gradients.empty()) { in XLAShapeForArgument()
664 arg.tensor_array_gradients.size() + 1, *xla_shape); in XLAShapeForArgument()
733 /*tensor_array_gradients=*/arg.tensor_array_gradients, in BuildArguments()
[all …]
Dxla_resource.cc66 const std::set<string>& tensor_array_gradients, in XlaResource() argument
80 for (const string& gradient : tensor_array_gradients) { in XlaResource()
Dxla_compiler.h160 std::set<string> tensor_array_gradients; member
Dxla_compiler_test.cc753 args[0].tensor_array_gradients = {"grad2"}; in TEST_F()
813 args[0].tensor_array_gradients = {"grad1"}; in TEST_F()
846 args[0].tensor_array_gradients = {"grad1"}; in TEST_F()