| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_filter_ops.cc | 108 Tensor* filter_backprop, TensorFormat data_format) { in operator ()()
111 d, filter_backprop->tensor<T, 4>(), input.tensor<T, 4>(), in operator ()()
256 Tensor* filter_backprop; in Compute() local
258 context->allocate_output(0, filter_shape, &filter_backprop)); in Compute()
284 filter_backprop->tensor<T, 4>(), out_backprop.tensor<T, 4>(), in Compute()
344 T* filter_backprop_data = filter_backprop->template flat<T>().data(); in Compute()
502 Tensor* filter_backprop = nullptr; in Compute() local
504 context->allocate_output(0, filter_shape, &filter_backprop)); in Compute()
513 f(context->eigen_device<Device>(), filter_backprop->flat<T>()); in Compute()
526 explicit_paddings_, filter_backprop, data_format_); in Compute()
[all …]
|
| D | depthwise_conv_op_gpu.h | 1038 T* filter_backprop,
1092 filter_backprop +
1116 filter_backprop +
1322 T* filter_backprop,
1380 filter_backprop +
1405 filter_backprop +
1587 const T* out_backprop, const T* input, T* filter_backprop,
1629 device.stream()>>>(args, out_backprop, input, filter_backprop);
1637 const T* out_backprop, const T* input, T* filter_backprop,
1643 ctx, args, block_height, out_backprop, input, filter_backprop,
[all …]
|
| D | dilation_ops.cc | 366 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 ()()
|
| D | depthwise_conv_grad_op.cc | 829 const T* out_backprop, const T* input, T* filter_backprop, in operator ()()
906 memset(filter_backprop, 0, filter_spatial_size * out_depth * sizeof(T)); in operator ()()
914 auto out_block_data = filter_backprop + output_base + j; in operator ()()
933 filter_backprop[output_base + vectorized_size + j] += in operator ()()
945 T* filter_backprop) { in DepthwiseConvBackpropFilterReference() argument
948 memset(filter_backprop, 0, num_filter_backprop * sizeof(T)); in DepthwiseConvBackpropFilterReference()
978 filter_backprop[filter_backprop_offset] += in DepthwiseConvBackpropFilterReference()
1075 Tensor* filter_backprop = nullptr; in Compute() local
1077 {1}, 0, filter_shape, &filter_backprop)); in Compute()
1111 context, reshaped_filter.CopyFrom(*filter_backprop, shape), in Compute()
[all …]
|
| D | conv_grad_ops.h | 191 Tensor* filter_backprop, TensorFormat data_format);
212 Tensor* filter_backprop, TensorFormat data_format);
|
| D | conv_grad_ops_3d.cc | 695 Tensor* filter_backprop; in Compute() local
697 context->allocate_output(0, filter_shape, &filter_backprop)); in Compute()
700 filter_backprop->template flat<T>().setZero(); in Compute()
706 filter_backprop->tensor<T, 5>(), // filter_backward in Compute()
807 Tensor* filter_backprop; in Compute() local
809 context->allocate_output(0, filter_shape, &filter_backprop)); in Compute()
812 filter_backprop->template flat<T>().setZero(); in Compute()
896 filter_backprop->tensor<T, 5>(), // filter_backward in Compute()
923 T* filter_backprop_data = filter_backprop->template flat<T>().data(); in Compute()
1545 Tensor* filter_backprop; in Compute() local
[all …]
|
| D | dilation_ops_gpu.cu.cc | 257 typename TTypes<T, 3>::Tensor filter_backprop) { in operator ()()
277 total_count, filter_backprop.data())); in operator ()()
287 rate_rows, rate_cols, pad_top, pad_left, filter_backprop.data()); in operator ()()
|
| D | depthwise_conv_op.h | 79 const T* out_backprop, const T* input, T* filter_backprop,
101 const T* out_backprop, const T* input, T* filter_backprop,
|
| D | dilation_ops.h | 60 typename TTypes<T, 3>::Tensor filter_backprop);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 147 const xla::XlaOp& filter_backprop, in ContractFilterForDepthwiseBackprop() argument
151 filter_backprop, xla::ZerosLike(filter_backprop)); in ContractFilterForDepthwiseBackprop()
405 xla::XlaOp filter_backprop; in MakeXlaBackpropFilterConvOp() local
532 filter_backprop = xla::ConvGeneralDilated( in MakeXlaBackpropFilterConvOp()
539 filter_backprop = ContractFilterForDepthwiseBackprop( in MakeXlaBackpropFilterConvOp()
540 filter_shape, filter_backprop, activations.builder()); in MakeXlaBackpropFilterConvOp()
543 return filter_backprop; in MakeXlaBackpropFilterConvOp()
|
| D | conv_ops.cc | 164 xla::StatusOr<xla::XlaOp> filter_backprop = MakeXlaBackpropFilterConvOp( in Compile() local
167 OP_REQUIRES_OK(ctx, filter_backprop.status()); in Compile()
168 ctx->SetOutput(0, filter_backprop.ValueOrDie()); in Compile()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Dilation2DBackpropFilter.pbtxt | 22 name: "filter_backprop"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 21056 name: "filter_backprop"
21114 name: "filter_backprop"
21174 name: "filter_backprop"
21235 name: "filter_backprop"
|
| D | ops_history.v1.pbtxt | 21399 name: "filter_backprop"
21457 name: "filter_backprop"
21517 name: "filter_backprop"
21578 name: "filter_backprop"
|
| D | ops_history.v0.pbtxt | 10531 name: "filter_backprop"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 29414 …t_backprop tf.Output, strides []int64, rates []int64, padding string) (filter_backprop tf.Output) {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 9861 name: "filter_backprop"
|