Searched refs:opdef (Results 1 – 7 of 7) sorted by relevance
31 func GetAPIDef(t *testing.T, opdef *odpb.OpDef, apidefText string) *adpb.ApiDef {32 opdefList := &odpb.OpList{Op: []*odpb.OpDef{opdef}}41 apidef, err := apimap.Get(opdef.Name)53 opdef string59 opdef: `85 opdef: `138 opdef: `184 opdef: `311 opdef: `355 opdef: `[all …]
64 const OpDef& opdef) -> Status { in LoadDynamicLibrary() argument66 if (seen_op_names.find(opdef.name()) == seen_op_names.end()) { in LoadDynamicLibrary()73 *library.op_list.add_op() = opdef; in LoadDynamicLibrary()74 seen_op_names.insert(opdef.name()); in LoadDynamicLibrary()
55 tensorflow::OpDef opdef; in RegisterCustomOps() local57 &opdef)) { in RegisterCustomOps()63 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomOps()64 *op_reg_data = tensorflow::OpRegistrationData(opdef); in RegisterCustomOps()
345 tensorflow::OpDef opdef; in RegisterCustomOpdefs() local346 if (!tensorflow::protobuf::TextFormat::ParseFromString(tf_opdefs, &opdef)) { in RegisterCustomOpdefs()356 [opdef]( in RegisterCustomOpdefs()358 *op_reg_data = tensorflow::OpRegistrationData(opdef); in RegisterCustomOpdefs()363 const char* node_name = opdef.name().c_str(); in RegisterCustomOpdefs()364 const char* op_name = opdef.name().c_str(); in RegisterCustomOpdefs()
163 tensorflow::OpDef opdef; in RegisterCustomBuiltinOps() local165 &opdef)) { in RegisterCustomBuiltinOps()170 tensorflow::OpRegistry::Global()->LookUp(opdef.name()); in RegisterCustomBuiltinOps()174 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterCustomBuiltinOps()175 *op_reg_data = tensorflow::OpRegistrationData(opdef); in RegisterCustomBuiltinOps()
79 tensorflow::OpDef opdef; in RegisterExtraTfOpDefs() local81 &opdef)) { in RegisterExtraTfOpDefs()88 [opdef](tensorflow::OpRegistrationData* op_reg_data) -> Status { in RegisterExtraTfOpDefs()89 *op_reg_data = tensorflow::OpRegistrationData(opdef); in RegisterExtraTfOpDefs()
357 const OpDef* opdef; in TEST() local358 Status nested_error = OpRegistry::Global()->LookUpOpDef("Zzzx", &opdef); in TEST()