Searched refs:y_grad_node_outputs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gradients.cc | 156 gtl::ArraySlice<NodeOut> y_grad_node_outputs,
213 gtl::ArraySlice<NodeOut> y_grad_node_outputs, in SymbolicGradientBuilder() argument
217 y_grad_node_outputs_(y_grad_node_outputs), in SymbolicGradientBuilder()
220 CHECK_EQ(y_node_outputs_.size(), y_grad_node_outputs.size()); in SymbolicGradientBuilder()
409 gtl::ArraySlice<NodeOut> y_grad_node_outputs, in AddSymbolicGradients() argument
413 y_grad_node_outputs, x_grad_node_outputs, in AddSymbolicGradients()
|
| D | gradients.h | 52 gtl::ArraySlice<NodeOut> y_grad_node_outputs,
|
| D | function.cc | 1454 std::vector<NodeOut> y_grad_node_outputs; in Compute() local
1455 y_grad_node_outputs.reserve(num_y); in Compute()
1465 y_grad_node_outputs.push_back({dy, 0}); in Compute()
1481 y_grad_node_outputs, &x_grad_node_outputs, in Compute()
|