Searched refs:tensor_array_gradients (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | if_op.cc | 77 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()
|
D | case_op.cc | 74 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()
|
D | while_op.cc | 75 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/ |
D | xla_resource.h | 53 const std::set<string>& tensor_array_gradients, 159 const std::map<string, std::unique_ptr<XlaResource>>& tensor_array_gradients() in tensor_array_gradients() function
|
D | xla_compiler.cc | 265 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 …]
|
D | xla_resource.cc | 66 const std::set<string>& tensor_array_gradients, in XlaResource() argument 80 for (const string& gradient : tensor_array_gradients) { in XlaResource()
|
D | xla_compiler.h | 160 std::set<string> tensor_array_gradients; member
|
D | xla_compiler_test.cc | 753 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()
|