Searched refs:reshaped_filter (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op.cc | 398 Tensor reshaped_filter(/*type=*/dtype_); in Compute() local
400 context, reshaped_filter.CopyFrom(filter, shape), in Compute()
406 reshaped_filter, /*row_dilation=*/1, /*col_dilation=*/1, in Compute()
|
| D | depthwise_conv_grad_op.cc | 627 Tensor reshaped_filter(/*type=*/dtype_); in Compute() local
629 context, reshaped_filter.CopyFrom(filter, shape), in Compute()
635 reshaped_filter, /*row_dilation=*/1, /*col_dilation=*/1, in Compute()
1109 Tensor reshaped_filter(/*type=*/dtype_); in Compute() local
1111 context, reshaped_filter.CopyFrom(*filter_backprop, shape), in Compute()
1119 padding_, /*explicit_paddings=*/{}, &reshaped_filter, in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 360 auto reshaped_filter = in HandleConvolution() local
370 expanded_filter_shape, reshaped_filter, broadcast_dims)); in HandleConvolution()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/ |
| D | depthwise_conv.cc | 154 int8* reshaped_filter = in DepthwiseConvOptimizedForFilterWidthEight() local
157 *reshaped_filter = (int32_t)(*current_filter) + filter_offset; in DepthwiseConvOptimizedForFilterWidthEight()
|