Home
last modified time | relevance | path

Searched defs:reshaped_filter (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.cc438 Tensor reshaped_filter(/*type=*/dtype_); in Compute() local
Ddepthwise_conv_grad_op.cc669 Tensor reshaped_filter(/*type=*/dtype_); in Compute() local
1166 Tensor reshaped_filter(/*type=*/dtype_); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc461 auto reshaped_filter = in HandleConvolution() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc301 auto reshaped_filter = rewriter.create<mhlo::ReshapeOp>( in CreateConvOp() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc2270 Value reshaped_filter = rewriter.create<linalg::TensorCollapseShapeOp>( in matchAndRewrite() local