Home
last modified time | relevance | path

Searched refs:SliceGrad (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Darray_grad.cc439 Status SliceGrad(const AttrSlice& attrs, FunctionDef* g) { in SliceGrad() function
472 REGISTER_OP_GRADIENT("Slice", SliceGrad);
Darray_grad_test.cc522 std::vector<Tensor> SliceGrad(const Tensor& x, const Tensor& b, const Tensor& s, in SliceGrad() function
546 TEST(ArrayGradTest, SliceGrad) { in TEST() argument
553 auto dx = SliceGrad(x, begin, size, dy); in TEST()
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc442 Status SliceGrad(const Scope& scope, const Operation& op, in SliceGrad() function
492 REGISTER_GRADIENT_OP("Slice", SliceGrad);
Darray_grad_test.cc389 TEST_F(ArrayGradTest, SliceGrad) { in TEST_F() argument