/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | pooling_test.cc | 188 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
|
D | pooling.cc | 192 XlaOp AvgPoolGrad(XlaOp out_backprop, absl::Span<const int64> gradients_size, in AvgPoolGrad()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | bias_op_test.cc | 28 Tensor out_backprop(DT_FLOAT, TensorShape({d0, d1, d2, d3})); in BiasAddGrad() local
|
D | avgpooling_op.cc | 244 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
|
D | mkl_conv_ops_test.cc | 179 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
|
D | conv_grad_ops_3d.cc | 220 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
|
D | conv_grad_filter_ops.cc | 238 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 ()()
|
D | batch_norm_op.cc | 102 const Tensor& out_backprop = context->input(4); in Compute() local
|
D | depthwise_conv_grad_op.cc | 183 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()
|
D | cudnn_pooling_gpu.cc | 123 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute()
|
D | dilation_ops.cc | 222 const Tensor& out_backprop = context->input(2); in Compute() local 342 const Tensor& out_backprop = context->input(2); in Compute() local
|
D | conv_grad_input_ops.cc | 316 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 ()()
|
D | mkl_avgpooling_op.cc | 246 const Tensor& out_backprop = MklGetInput(context, 1); in Compute() local 354 const Tensor& out_backprop = MklGetInput(context, 1); in MklCreateLayoutsAndPrimitives() local
|
D | maxpooling_op.cc | 60 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
|
D | pooling_ops_common.cc | 289 const Tensor* tensor_out, const Tensor& out_backprop, in Compute()
|
D | fractional_avg_pool_op.cc | 240 const Tensor& out_backprop = context->input(1); in Compute() local
|
D | pooling_ops_3d.cc | 342 const Tensor& out_backprop = context->input(2); in Compute() local 495 const Tensor& out_backprop = context->input(1); in Compute() local
|
D | fractional_max_pool_op.cc | 232 const Tensor& out_backprop = context->input(2); in Compute() local
|
D | mkl_maxpooling_op.cc | 392 const Tensor& out_backprop = MklGetInput(context, 2); in MklPrepareInputs() local
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | pooling_ops.cc | 327 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
|
D | conv_op_helpers.cc | 325 xla::XlaOp out_backprop, const ConvOpAttrs& attrs) { in MakeXlaBackpropInputConvOp()
|
/external/tensorflow/tensorflow/python/ops/ |
D | nn_ops.py | 1958 out_backprop, argument 2018 out_backprop=None, argument
|