Home
last modified time | relevance | path

Searched defs:out_backprop (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling_test.cc188 XlaOp out_backprop = ConstantR4FromArray4D<float>(&builder, {{{{1.}}}}); in XLA_TEST_F() local
205 XlaOp out_backprop = in XLA_TEST_F() local
223 XlaOp out_backprop = in XLA_TEST_F() local
240 XlaOp out_backprop = in XLA_TEST_F() local
255 XlaOp out_backprop = in XLA_TEST_F() local
274 XlaOp out_backprop = in XLA_TEST_F() local
Dpooling.cc193 XlaOp AvgPoolGrad(XlaOp out_backprop, absl::Span<const int64> gradients_size, in AvgPoolGrad()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops_test.cc165 Node* out_backprop = in DefaultConv2DBwdInput() local
193 Node* out_backprop = in MklConv2DBwdInput() local
229 Node* out_backprop = in DefaultConv2DBwdFilter() local
258 Node* out_backprop = in MklConv2DBwdFilter() local
/external/tensorflow/tensorflow/core/kernels/
Davgpooling_op.cc278 const Tensor& out_backprop = context->input(1); in Compute() local
440 const Tensor& out_backprop = context->input(1); in Compute() local
527 const Tensor& out_backprop = context->input(1); in Compute() local
Dbias_op_test.cc40 Tensor out_backprop(DT_FLOAT, TensorShape({d0, d1, d2, d3})); in BiasAddGrad() local
Dconv_grad_ops_3d.cc229 const Tensor& out_backprop = context->input(2); in Compute() local
336 const Tensor& out_backprop = context->input(2); in Compute() local
687 const Tensor& out_backprop = context->input(2); in Compute() local
799 const Tensor& out_backprop = context->input(2); in Compute() local
1155 const Tensor& out_backprop = context->input(2); in Compute() local
1637 const Tensor& out_backprop = context->input(2); in Compute() local
Ddepthwise_conv_grad_op.cc195 const T* out_backprop, T* buffer) { in CopyOutputBackpropRegion()
473 const T* out_backprop, in DepthwiseConvBackpropInputReference()
790 const T* out_backprop, const T* input_buffer, in ComputeBackpropFilter()
984 const T* out_backprop, in DepthwiseConvBackpropFilterReference()
Dbatch_norm_op.cc99 const Tensor& out_backprop = context->input(4); in Compute() local
Dconv_grad_filter_ops.cc322 const Tensor& out_backprop = context->input(2); in Compute() local
430 const Tensor& out_backprop = context->input(2); in Compute() local
652 const Tensor& out_backprop, const Tensor& input, int row_dilation, in operator ()()
Ddilation_ops.cc222 const Tensor& out_backprop = context->input(2); in Compute() local
342 const Tensor& out_backprop = context->input(2); in Compute() local
Dcudnn_pooling_gpu.cc138 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute()
Dpooling_ops_3d.cc338 const Tensor& out_backprop = context->input(2); in Compute() local
491 const Tensor& out_backprop = context->input(1); in Compute() local
Dfractional_avg_pool_op.cc240 const Tensor& out_backprop = context->input(1); in Compute() local
Dfractional_max_pool_op.cc232 const Tensor& out_backprop = context->input(2); in Compute() local
Dmaxpooling_op.cc64 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, in SpatialMaxPoolWithArgMaxHelper()
265 const Tensor& out_backprop = context->input(2); in Compute() local
383 const Tensor& out_backprop = context->input(2); in Compute() local
Dconv_grad_input_ops.cc68 const Tensor& out_backprop, const Tensor& filter, int row_dilation, in operator ()()
Dpooling_ops_common.cc446 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc334 auto out_backprop = ctx->Input(2); in Compile() local
440 auto out_backprop = ctx->Input(1); in Compile() local
576 auto out_backprop = ctx->Input(2); in Compile() local
Dconv_op_helpers.cc295 xla::XlaOp out_backprop, const ConvOpAttrs& attrs, in MakeXlaBackpropInputConvOp()
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py2299 out_backprop, argument
2360 out_backprop=None, argument
2862 out_backprop, argument
2932 out_backprop, argument
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc202 auto out_backprop = in CreateSimpleConv2DBackpropFilter() local
241 auto out_backprop = in CreateSimpleConv2DBackpropInput() local