Searched refs:tensor_array_gradients (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_compiler.cc | 294 for (const auto& grad : resource->tensor_array_gradients()) { in BuildComputation()
298 arg.tensor_array_gradients.count(grad.first) == 0; in BuildComputation()
324 for (const auto& grad : resource->tensor_array_gradients()) { in BuildComputation()
458 if (!tensor_array_gradients.empty()) { in HumanString()
460 absl::StrJoin(tensor_array_gradients, ",")); in HumanString()
923 if (!arg.tensor_array_gradients.empty()) { in XLAShapeForArgument()
925 arg.tensor_array_gradients.size() + 1, *xla_shape); in XLAShapeForArgument()
971 for (const auto& gradient : resource.tensor_array_gradients()) { in PopulateArgumentFromResource()
972 arg->tensor_array_gradients.insert(gradient.first); in PopulateArgumentFromResource()
1011 /*tensor_array_gradients=*/arg.tensor_array_gradients, in BuildArguments()
[all …]
|
| H A D | xla_argument.cc | 24 tensor_array_gradients) != in operator ==()
27 other.tensor_array_gradients)) { in operator ==()
|
| H A D | xla_resource.h | 54 const std::set<string>& tensor_array_gradients,
164 const std::map<string, std::unique_ptr<XlaResource>>& tensor_array_gradients() in tensor_array_gradients() function
|
| H A D | xla_resource.cc | 65 const std::set<string>& tensor_array_gradients, in XlaResource() argument
81 for (const string& gradient : tensor_array_gradients) { in XlaResource()
|
| H A D | xla_argument.h | 107 std::set<string> tensor_array_gradients; member
|
| H A D | xla_compiler_test.cc | 828 args[0].tensor_array_gradients = {"grad2"}; in TEST_F()
886 args[0].tensor_array_gradients = {"grad1"}; in TEST_F()
918 args[0].tensor_array_gradients = {"grad1"}; in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | case_op.cc | 190 for (const auto& gradient : resource->tensor_array_gradients()) { in Compile()
191 arg.tensor_array_gradients.insert(gradient.first); in Compile()
193 if (!resource->tensor_array_gradients().empty()) { in Compile()
351 arguments[update.input_index].tensor_array_gradients, in Compile()
|
| H A D | if_op.cc | 79 for (const auto& gradient : resource->tensor_array_gradients()) { in PopulateTensorArrayGradients()
80 arg.tensor_array_gradients.insert(gradient.first); in PopulateTensorArrayGradients()
82 if (!resource->tensor_array_gradients().empty()) in PopulateTensorArrayGradients()
369 arguments[update.input_index].tensor_array_gradients, in Compile()
|
| H A D | while_op.cc | 411 for (const auto& gradient : resource->tensor_array_gradients()) { in Compile()
412 arg.tensor_array_gradients.insert(gradient.first); in Compile()
628 arguments[update.input_index].tensor_array_gradients, in Compile()
|