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.cc193 XlaOp AvgPoolGrad(XlaOp out_backprop, absl::Span<const int64_t> gradients_size, in AvgPoolGrad()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops_test.cc160 Node* out_backprop = in DefaultConv2DBwdInput() local
187 Node* out_backprop = in MklConv2DBwdInput() local
222 Node* out_backprop = in DefaultConv2DBwdFilter() local
250 Node* out_backprop = in MklConv2DBwdFilter() local
/external/tensorflow/tensorflow/core/kernels/
Davgpooling_op.cc289 const Tensor& out_backprop = context->input(1); in Compute() local
452 const Tensor& out_backprop = context->input(1); in Compute() local
539 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.cc233 const Tensor& out_backprop = context->input(2); in Compute() local
362 const Tensor& out_backprop = context->input(2); in Compute() local
740 const Tensor& out_backprop = context->input(2); in Compute() local
878 const Tensor& out_backprop = context->input(2); in Compute() local
1267 const Tensor& out_backprop = context->input(2); in Compute() local
1647 const Tensor& out_backprop = context->input(2); in Compute() local
Ddepthwise_conv_grad_op.cc196 const T* out_backprop, T* buffer) { in CopyOutputBackpropRegion()
477 const T* out_backprop, in DepthwiseConvBackpropInputReference()
794 const T* out_backprop, const T* input_buffer, in ComputeBackpropFilter()
988 const T* out_backprop, in DepthwiseConvBackpropFilterReference()
Dconv_grad_filter_ops.cc326 const Tensor& out_backprop = context->input(2); in Compute() local
434 const Tensor& out_backprop = context->input(2); in Compute() local
671 const Tensor& out_backprop, const Tensor& input, int row_dilation, in operator ()()
Dbatch_norm_op.cc99 const Tensor& out_backprop = context->input(4); in Compute() local
Ddilation_ops.cc224 const Tensor& out_backprop = context->input(2); in Compute() local
357 const Tensor& out_backprop = context->input(2); in Compute() local
Dcudnn_pooling_gpu.cc133 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute()
Dpooling_ops_3d.cc339 const Tensor& out_backprop = context->input(2); in Compute() local
514 const Tensor& out_backprop = context->input(1); in Compute() local
Dfractional_max_pool_op.cc241 const Tensor& out_backprop = context->input(2); in Compute() local
Dconv_grad_input_ops.cc75 const Tensor& out_backprop, const Tensor& filter, int row_dilation, in operator ()()
Dfractional_avg_pool_op.cc261 const Tensor& out_backprop = context->input(1); in Compute() local
Dmaxpooling_op.cc65 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, in SpatialMaxPoolWithArgMaxHelper()
269 const Tensor& out_backprop = context->input(2); in Compute() local
395 const Tensor& out_backprop = context->input(2); in Compute() local
Dpooling_ops_common.cc445 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc382 auto out_backprop = ctx->Input(2); in Compile() local
488 auto out_backprop = ctx->Input(1); in Compile() local
628 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/c/experimental/ops/
Dnn_ops.cc127 AbstractTensorHandle* const out_backprop, in BiasAddGrad()
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py2488 out_backprop, argument
2549 out_backprop=None, argument
3059 out_backprop, argument
3131 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