Searched refs:MapSizeOp (Results 1 – 1 of 1) sorted by relevance
756 class MapSizeOp : public OpKernel { class758 explicit MapSizeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in MapSizeOp() function in tensorflow::__anon07039e130111::MapSizeOp774 REGISTER_KERNEL_BUILDER(Name("MapSize").Device(DEVICE_CPU), MapSizeOp<false>);776 MapSizeOp<true>);780 MapSizeOp<false>);783 MapSizeOp<true>);787 MapSizeOp<false>);790 MapSizeOp<true>);