Home
last modified time | relevance | path

Searched refs:filter_backprop (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DDilation2DBackpropFilter.pbtxt16 name: "filter_backprop"
74 name: "filter_backprop"
134 name: "filter_backprop"
195 name: "filter_backprop"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DDilation2DBackpropFilter.pbtxt16 name: "filter_backprop"
74 name: "filter_backprop"
134 name: "filter_backprop"
195 name: "filter_backprop"
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_filter_ops.cc113 Tensor* filter_backprop, TensorFormat data_format) { in operator ()()
121 TensorShape filter_shape = filter_backprop->shape(); in operator ()()
158 auto filter_backprop_t = filter_backprop->tensor<T, 4>(); in operator ()()
332 Tensor* filter_backprop = nullptr; in Compute() local
334 context->allocate_output(0, filter_shape, &filter_backprop)); in Compute()
343 f(context->eigen_device<Device>(), filter_backprop->flat<T>()); in Compute()
363 explicit_paddings_, filter_backprop, data_format_); in Compute()
449 Tensor* filter_backprop; in Compute() local
451 context->allocate_output(0, filter_shape, &filter_backprop)); in Compute()
483 filter_backprop->tensor<T, 4>(), out_backprop.tensor<T, 4>(), in Compute()
[all …]
Ddepthwise_conv_op_gpu.h1032 const T* __restrict__ input, T* __restrict__ filter_backprop,
1086 filter_backprop +
1110 filter_backprop +
1317 const T* __restrict__ input, T* __restrict__ filter_backprop,
1375 filter_backprop +
1400 filter_backprop +
1584 const T* out_backprop, const T* input, T* filter_backprop,
1626 out_backprop, input, filter_backprop));
1634 const T* out_backprop, const T* input, T* filter_backprop,
1642 ctx, args, block_height, out_backprop, input, filter_backprop,
[all …]
Dconv_grad_ops.h189 Tensor* filter_backprop, TensorFormat data_format);
210 Tensor* filter_backprop, TensorFormat data_format);
Ddilation_ops.cc366 Tensor* filter_backprop = nullptr; in Compute() local
368 context, context->allocate_output(0, filter.shape(), &filter_backprop)); in Compute()
379 filter_backprop->tensor<T, 3>()); in Compute()
396 typename TTypes<T, 3>::Tensor filter_backprop) { in operator ()()
409 filter_backprop.setZero(); in operator ()()
441 filter_backprop(h_max, w_max, d) += in operator ()()
Ddepthwise_conv_grad_op.cc870 const T* out_backprop, const T* input, T* filter_backprop, in operator ()()
947 memset(filter_backprop, 0, filter_spatial_size * out_depth * sizeof(T)); in operator ()()
955 auto out_block_data = filter_backprop + output_base + j; in operator ()()
974 filter_backprop[output_base + vectorized_size + j] += in operator ()()
986 T* filter_backprop) { in DepthwiseConvBackpropFilterReference() argument
989 memset(filter_backprop, 0, num_filter_backprop * sizeof(T)); in DepthwiseConvBackpropFilterReference()
1019 filter_backprop[filter_backprop_offset] += in DepthwiseConvBackpropFilterReference()
1125 Tensor* filter_backprop = nullptr; in Compute() local
1127 {1}, 0, filter_shape, &filter_backprop)); in Compute()
1167 context, reshaped_filter.CopyFrom(*filter_backprop, shape), in Compute()
[all …]
Dconv_grad_ops_3d.cc706 Tensor* filter_backprop; in Compute() local
708 context->allocate_output(0, filter_shape, &filter_backprop)); in Compute()
711 filter_backprop->template flat<T>().setZero(); in Compute()
717 filter_backprop->tensor<T, 5>(), // filter_backward in Compute()
818 Tensor* filter_backprop; in Compute() local
820 context->allocate_output(0, filter_shape, &filter_backprop)); in Compute()
823 filter_backprop->template flat<T>().setZero(); in Compute()
907 filter_backprop->tensor<T, 5>(), // filter_backward in Compute()
934 T* filter_backprop_data = filter_backprop->template flat<T>().data(); in Compute()
1656 Tensor* filter_backprop; in Compute() local
[all …]
Ddilation_ops_gpu.cu.cc257 typename TTypes<T, 3>::Tensor filter_backprop) { in operator ()()
277 total_count, filter_backprop.data())); in operator ()()
288 filter_backprop.data())); in operator ()()
Ddilation_ops.h60 typename TTypes<T, 3>::Tensor filter_backprop);
Ddepthwise_conv_op.h79 const T* out_backprop, const T* input, T* filter_backprop,
101 const T* out_backprop, const T* input, T* filter_backprop,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_ops.cc177 xla::StatusOr<xla::XlaOp> filter_backprop = MakeXlaBackpropFilterConvOp( in Compile() local
180 OP_REQUIRES_OK(ctx, filter_backprop.status()); in Compile()
181 ctx->SetOutput(0, filter_backprop.ValueOrDie()); in Compile()
Dconv_op_helpers.cc402 xla::XlaOp filter_backprop; in MakeXlaBackpropFilterConvOp() local
529 filter_backprop = xla::DynamicConvKernelGrad( in MakeXlaBackpropFilterConvOp()
536 filter_backprop = xla::ConvGeneralDilated( in MakeXlaBackpropFilterConvOp()
544 filter_backprop = xla::Reshape(filter_backprop, filter_shape.dimensions()); in MakeXlaBackpropFilterConvOp()
547 return filter_backprop; in MakeXlaBackpropFilterConvOp()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Dilation2DBackpropFilter.pbtxt22 name: "filter_backprop"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt21056 name: "filter_backprop"
21114 name: "filter_backprop"
21174 name: "filter_backprop"
21235 name: "filter_backprop"
Dops_history.v0.pbtxt10531 name: "filter_backprop"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go25056 …t_backprop tf.Output, strides []int64, rates []int64, padding string) (filter_backprop tf.Output) {
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt13185 name: "filter_backprop"