Searched refs:reshaped_filter (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op.cc | 437 Tensor reshaped_filter(/*type=*/dtype_); in Compute() local
439 context, reshaped_filter.CopyFrom(filter, shape), in Compute()
445 reshaped_filter, /*row_dilation=*/1, /*col_dilation=*/1, in Compute()
|
| D | depthwise_conv_grad_op.cc | 668 Tensor reshaped_filter(/*type=*/dtype_); in Compute() local
670 context, reshaped_filter.CopyFrom(filter, shape), in Compute()
676 reshaped_filter, /*row_dilation=*/1, /*col_dilation=*/1, in Compute()
1165 Tensor reshaped_filter(/*type=*/dtype_); in Compute() local
1167 context, reshaped_filter.CopyFrom(*filter_backprop, shape), in Compute()
1176 padding_, explicit_paddings_, &reshaped_filter, data_format_); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 460 auto reshaped_filter = in HandleConvolution() local
470 expanded_filter_shape, reshaped_filter, broadcast_dims)); in HandleConvolution()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 173 auto reshaped_filter = rewriter.create<mhlo::ReshapeOp>( in CreateConvOp() local
179 conv_op, conv_op.getType(), conv_op.lhs(), reshaped_filter, in CreateConvOp()
|