Searched refs:ConcatGradV2 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad.cc | 220 Status ConcatGradV2(const AttrSlice& attrs, FunctionDef* g) { in ConcatGradV2() function
225 REGISTER_OP_GRADIENT("ConcatV2", ConcatGradV2);
|
| D | array_grad_test.cc | 141 std::vector<Tensor> ConcatGradV2(int dim, const Tensor& x0, const Tensor& x1, in ConcatGradV2() function
187 dx = ConcatGradV2(1, x0, x1, dy); in TEST()
200 dx = ConcatGradV2(-2, x0, x1, dy); in TEST()
|