Searched refs:odpb (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/go/genop/internal/ |
| D | genop.go | 51 odpb "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/op_def_go_proto" packageName
73 func registeredOps() (*odpb.OpList, *apiDefMap, error) {
77 list = new(odpb.OpList)
111 func generateFunctionsForOps(w io.Writer, ops *odpb.OpList, apimap *apiDefMap) error {
136 func generateFunctionForOp(w io.Writer, op *odpb.OpDef, apidef *adpb.ApiDef) error {
362 op *odpb.OpDef_AttrDef
376 op *odpb.OpDef_ArgDef
386 Op *odpb.OpDef
401 func newTmplArgs(op *odpb.OpDef, apidef *adpb.ApiDef) (*tmplArgs, error) {
559 func isListArg(argdef *odpb.OpDef_ArgDef) bool {
[all …]
|
| D | genop_test.go | 26 odpb "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/op_def_go_proto" packageName
31 func GetAPIDef(t *testing.T, opdef *odpb.OpDef, apidefText string) *adpb.ApiDef {
32 opdefList := &odpb.OpList{Op: []*odpb.OpDef{opdef}}
797 var opdef odpb.OpDef
|
| D | api_def_map.go | 35 odpb "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/op_def_go_proto" packageName
54 func newAPIDefMap(oplist *odpb.OpList) (*apiDefMap, error) {
|