Searched refs:tensor_array_gradients (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_argument.cc | 24 tensor_array_gradients) != in operator ==()
27 other.tensor_array_gradients)) { in operator ==()
|
| D | xla_resource.h | 53 const std::set<string>& tensor_array_gradients,
161 const std::map<string, std::unique_ptr<XlaResource>>& tensor_array_gradients() in tensor_array_gradients() function
|
| D | xla_compiler.cc | 291 for (const auto& grad : resource->tensor_array_gradients()) { in BuildComputation()
295 arg.tensor_array_gradients.count(grad.first) == 0; in BuildComputation()
320 for (const auto& grad : resource->tensor_array_gradients()) { in BuildComputation()
459 if (!tensor_array_gradients.empty()) { in HumanString()
461 absl::StrJoin(tensor_array_gradients, ",")); in HumanString()
897 if (!arg.tensor_array_gradients.empty()) { in XLAShapeForArgument()
899 arg.tensor_array_gradients.size() + 1, *xla_shape); in XLAShapeForArgument()
945 for (const auto& gradient : resource.tensor_array_gradients()) { in PopulateArgumentFromResource()
946 arg->tensor_array_gradients.insert(gradient.first); in PopulateArgumentFromResource()
985 /*tensor_array_gradients=*/arg.tensor_array_gradients, in BuildArguments()
[all …]
|
| D | xla_argument.h | 104 std::set<string> tensor_array_gradients; member
|
| D | xla_resource.cc | 65 const std::set<string>& tensor_array_gradients, in XlaResource() argument
79 for (const string& gradient : tensor_array_gradients) { in XlaResource()
|
| D | xla_compiler_test.cc | 822 args[0].tensor_array_gradients = {"grad2"}; in TEST_F()
881 args[0].tensor_array_gradients = {"grad1"}; in TEST_F()
913 args[0].tensor_array_gradients = {"grad1"}; in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | case_op.cc | 188 for (const auto& gradient : resource->tensor_array_gradients()) { in Compile()
189 arg.tensor_array_gradients.insert(gradient.first); in Compile()
191 if (!resource->tensor_array_gradients().empty()) { in Compile()
359 arguments[update.input_index].tensor_array_gradients, in Compile()
|
| D | if_op.cc | 76 for (const auto& gradient : resource->tensor_array_gradients()) { in PopulateTensorArrayGradients()
77 arg.tensor_array_gradients.insert(gradient.first); in PopulateTensorArrayGradients()
79 if (!resource->tensor_array_gradients().empty()) in PopulateTensorArrayGradients()
344 arguments[update.input_index].tensor_array_gradients, in Compile()
|
| D | while_op.cc | 399 for (const auto& gradient : resource->tensor_array_gradients()) { in Compile()
400 arg.tensor_array_gradients.insert(gradient.first); in Compile()
614 arguments[update.input_index].tensor_array_gradients, in Compile()
|