Home
last modified time | relevance | path

Searched refs:output_kernel (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_op_interfaces.h40 StringRef output_kernel, ArrayRef<int> additional_arguments = {},
42 : output_kernel(output_kernel.str()),
49 std::string output_kernel; member
/external/tensorflow/tensorflow/core/kernels/
Dconv_2d.h34 const OutputKernel& output_kernel,
42 row_dilation, output_kernel, padding_left, padding_right, padding_top,
54 const OutputKernel& output_kernel = OutputKernel()) { in operator()
56 row_dilation, col_dilation, padding, output_kernel); in operator()
64 const OutputKernel& output_kernel = OutputKernel()) { in operator()
67 col_dilation, Eigen::PaddingType::PADDING_VALID, output_kernel, in operator()
80 const OutputKernel& output_kernel = OutputKernel()) {
84 row_dilation, output_kernel)
94 const OutputKernel& output_kernel = OutputKernel()) {
99 output_kernel, padding_left, padding_right, padding_top,
[all …]
Dmatmul_op_fused.cc75 auto executeWithOutputKernel = [&](auto output_kernel) { in operator ()() argument
77 [&output_kernel]( in operator ()()
81 output_kernel(output_mapper, params, i, j, num_rows, num_cols); in operator ()()
Dconv_ops_fused_impl.h107 void operator()(const OutputKernel& output_kernel, OpKernelContext* ctx, in operator()
112 [&output_kernel]( in operator()
116 output_kernel(output_mapper, params, i, j, num_rows, num_cols); in operator()
Deigen_spatial_convolutions-inl.h1604 const OutputKernel& output_kernel = OutputKernel(),
1738 contract_dims, output_kernel)
1748 .contract(kernel.reshape(kernel_dims), contract_dims, output_kernel)
1759 contract_dims, output_kernel)
1766 .contract(kernel.reshape(kernel_dims), contract_dims, output_kernel)
Dconv_ops.cc1143 const Eigen::NoOpOutputKernel& output_kernel); \
1151 const Eigen::NoOpOutputKernel& output_kernel); \
1159 const Eigen::NoOpOutputKernel& output_kernel); \
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcontraction_fusion.cc103 StringAttr fusion_name = StringAttr::get(ctx, fusion->output_kernel); in matchAndRewrite()