Searched refs:AvgPool3DOp (Results 1 – 3 of 3) sorted by relevance
260 class AvgPool3DOp : public AvgPoolOp { class262 explicit AvgPool3DOp(OpKernelConstruction* ctx) in AvgPool3DOp() function in tensorflow::__anon9297e9860111::AvgPool3DOp265 REGISTER_XLA_OP(Name("AvgPool3D"), AvgPool3DOp);
936 rewriter.replaceOpWithNewOp<AvgPool3DOp>( in replaceWithAvgPool()
2250 Value GetAvgPoolInput(TF::AvgPool3DOp op) { return op.input(); } in GetAvgPoolInput()2320 using ConvertAvgPool3DOp = ConvertAvgPoolOp<TF::AvgPool3DOp, /*num_dims=*/5>;