Home
last modified time | relevance | path

Searched refs:StridedSliceGrad (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op_gpu_impl.h41 template struct functor::StridedSliceGrad<GPUDevice, T, 1>; \
42 template struct functor::StridedSliceGrad<GPUDevice, T, 2>; \
43 template struct functor::StridedSliceGrad<GPUDevice, T, 3>; \
44 template struct functor::StridedSliceGrad<GPUDevice, T, 4>; \
45 template struct functor::StridedSliceGrad<GPUDevice, T, 5>; \
46 template struct functor::StridedSliceGrad<GPUDevice, T, 6>; \
47 template struct functor::StridedSliceGrad<GPUDevice, T, 7>; \
Dstrided_slice_op_impl.h135 functor::StridedSliceGrad<Device, Proxy, NDIM>()( in HandleStridedSliceGradCase()
206 void StridedSliceGrad<GPUDevice, T, NDIM>::operator()( \
212 extern template struct StridedSliceGrad<GPUDevice, T, NDIM>; \
Dstrided_slice_op.h79 struct StridedSliceGrad {
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_StridedSliceGrad.pbtxt2 graph_op_name: "StridedSliceGrad"
10 Arguments are the same as StridedSliceGrad with the exception that
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_StridedSliceGrad.pbtxt2 graph_op_name: "StridedSliceGrad"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_StridedSliceGrad.pbtxt2 graph_op_name: "StridedSliceGrad"
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc565 std::vector<Tensor> StridedSliceGrad(const Tensor& x, const Tensor& begin, in StridedSliceGrad() function
653 TEST(ArrayGradTest, StridedSliceGrad) { in TEST() argument
667 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask, in TEST()
694 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask, in TEST()
722 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask, in TEST()
750 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask, in TEST()
Darray_grad.cc474 Status StridedSliceGrad(const AttrSlice& attrs, FunctionDef* g) { in StridedSliceGrad() function
509 REGISTER_OP_GRADIENT("StridedSlice", StridedSliceGrad);
Dops.pbtxt39163 name: "StridedSliceGrad"
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc428 StridedSliceGrad(scope, x, begin, end, strides, grad_inputs[0], in StridedSliceGradHelper()
429 StridedSliceGrad::BeginMask(begin_mask) in StridedSliceGradHelper()
Darray_grad_test.cc365 TEST_F(ArrayGradTest, StridedSliceGrad) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dgpu_supported_ops.md238 `StridedSliceGrad` | `Index={int32,int64}`<br>`T={bool,complex64,double,float,in…
Dcpu_supported_ops.md241 `StridedSliceGrad` | `Index={int32,int64}`<br>`T={bool,complex64,double,float,in…
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3247 TEST_F(OpTest, StridedSliceGrad) { in TEST_F() argument
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt3748 name: "StridedSliceGrad"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt3748 name: "StridedSliceGrad"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go15141 func StridedSliceGrad(scope *Scope, shape tf.Output, begin tf.Output, end tf.Output, strides tf.Out… func
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt35763 name: "StridedSliceGrad"
Dops_history.v1.pbtxt81572 name: "StridedSliceGrad"
Dops_history.v2.pbtxt80945 name: "StridedSliceGrad"