Home
last modified time | relevance | path

Searched defs:out_backprop (Results 1 – 22 of 22) 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.cc192 XlaOp AvgPoolGrad(XlaOp out_backprop, absl::Span<const int64> gradients_size, in AvgPoolGrad()
/external/tensorflow/tensorflow/core/kernels/
Dbias_op_test.cc28 Tensor out_backprop(DT_FLOAT, TensorShape({d0, d1, d2, d3})); in BiasAddGrad() local
Davgpooling_op.cc244 const Tensor& out_backprop = context->input(1); in Compute() local
403 const Tensor& out_backprop = context->input(1); in Compute() local
483 const Tensor& out_backprop = context->input(1); in Compute() local
Dmkl_conv_ops_test.cc179 Node* out_backprop = in DefaultConv2DBwdInput() local
207 Node* out_backprop = in MklConv2DBwdInput() local
242 Node* out_backprop = in DefaultConv2DBwdFilter() local
271 Node* out_backprop = in MklConv2DBwdFilter() local
Dconv_grad_ops_3d.cc220 const Tensor& out_backprop = context->input(2); in Compute() local
326 const Tensor& out_backprop = context->input(2); in Compute() local
676 const Tensor& out_backprop = context->input(2); in Compute() local
788 const Tensor& out_backprop = context->input(2); in Compute() local
1142 const Tensor& out_backprop = context->input(2); in Compute() local
1526 const Tensor& out_backprop = context->input(2); in Compute() local
Dconv_grad_filter_ops.cc238 const Tensor& out_backprop = context->input(2); in Compute() local
492 const Tensor& out_backprop = context->input(2); in Compute() local
545 const Tensor& out_backprop, const Tensor& input, int row_dilation, in operator ()()
Dbatch_norm_op.cc102 const Tensor& out_backprop = context->input(4); in Compute() local
Ddepthwise_conv_grad_op.cc183 const T* out_backprop, T* buffer) { in CopyOutputBackpropRegion()
461 const T* out_backprop, in DepthwiseConvBackpropInputReference()
749 const T* out_backprop, const T* input_buffer, in ComputeBackpropFilter()
943 const T* out_backprop, in DepthwiseConvBackpropFilterReference()
Dcudnn_pooling_gpu.cc123 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute()
Ddilation_ops.cc222 const Tensor& out_backprop = context->input(2); in Compute() local
342 const Tensor& out_backprop = context->input(2); in Compute() local
Dconv_grad_input_ops.cc316 const Tensor& out_backprop = context->input(2); in Compute() local
635 const Tensor& out_backprop = context->input(2); in Compute() local
682 const Tensor& out_backprop, const Tensor& filter, int row_dilation, in operator ()()
Dmkl_avgpooling_op.cc246 const Tensor& out_backprop = MklGetInput(context, 1); in Compute() local
354 const Tensor& out_backprop = MklGetInput(context, 1); in MklCreateLayoutsAndPrimitives() local
Dmaxpooling_op.cc60 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, in SpatialMaxPoolWithArgMaxHelper()
249 const Tensor& out_backprop = context->input(2); in Compute() local
326 const Tensor& out_backprop, const TensorShape& tensor_in_shape) { in MaxPoolingBackwardCustomKernel()
381 const Tensor& out_backprop = context->input(2); in Compute() local
Dpooling_ops_common.cc289 const Tensor* tensor_out, const Tensor& out_backprop, in Compute()
Dfractional_avg_pool_op.cc240 const Tensor& out_backprop = context->input(1); in Compute() local
Dpooling_ops_3d.cc342 const Tensor& out_backprop = context->input(2); in Compute() local
495 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
Dmkl_maxpooling_op.cc392 const Tensor& out_backprop = MklGetInput(context, 2); in MklPrepareInputs() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc327 auto out_backprop = ctx->Input(2); in Compile() local
419 auto out_backprop = ctx->Input(1); in Compile() local
556 auto out_backprop = ctx->Input(2); in Compile() local
Dconv_op_helpers.cc325 xla::XlaOp out_backprop, const ConvOpAttrs& attrs) { in MakeXlaBackpropInputConvOp()
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py1958 out_backprop, argument
2018 out_backprop=None, argument