Searched refs:GetAttrKernel (Results 1 – 1 of 1) sorted by relevance
681 class GetAttrKernel : public ::tensorflow::OpKernel { class683 explicit GetAttrKernel(OpKernelConstruction* context) : OpKernel(context) { in GetAttrKernel() function in tensorflow::__anon843475bc0111::GetAttrKernel750 GetAttrKernel);756 auto* get_attr_kernel = static_cast<GetAttrKernel*>(op_kernel.get()); in TEST_F()762 get_attr_kernel = static_cast<GetAttrKernel*>(op_kernel.get()); in TEST_F()771 REGISTER_KERNEL_BUILDER(Name("GetAttrInt").Device(DEVICE_CPU), GetAttrKernel);777 auto* get_attr_kernel = static_cast<GetAttrKernel*>(op_kernel.get()); in TEST_F()785 get_attr_kernel = static_cast<GetAttrKernel*>(op_kernel.get()); in TEST_F()794 get_attr_kernel = static_cast<GetAttrKernel*>(op_kernel.get()); in TEST_F()808 get_attr_kernel = static_cast<GetAttrKernel*>(op_kernel.get()); in TEST_F()[all …]