Home
last modified time | relevance | path

Searched defs:OpContext (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_context.h28 struct OpContext { struct
38 OpContext() { function_library = nullptr; } in OpContext() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dsplit_v.cc33 struct OpContext { struct
34 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
40 TfLiteSplitVParams* params;
41 const TfLiteTensor* input;
42 const TfLiteTensor* size_splits;
43 const TfLiteTensor* axis;
H A Dsplit.cc31 struct OpContext { struct
32 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
37 TfLiteSplitParams* params;
38 const TfLiteTensor* axis;
39 const TfLiteTensor* input;
H A Dmaximum_minimum.cc44 struct OpContext { struct
45 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
50 const TfLiteTensor* input1;
51 const TfLiteTensor* input2;
52 TfLiteTensor* output;
H A Dfake_quant.cc33 struct OpContext { struct
34 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
38 const TfLiteTensor* input;
39 TfLiteTensor* output;
H A Ddensify.cc30 struct OpContext { struct
31 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
35 const TfLiteTensor* input;
36 TfLiteTensor* output;
H A Dnumeric_verify.cc45 struct OpContext { struct
46 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
51 const TfLiteTensor* input;
52 const TfLiteTensor* ref;
53 TfLiteTensor* output;
H A Ddequantize.cc28 struct OpContext { struct
29 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
33 const TfLiteTensor* input;
34 TfLiteTensor* output;
H A Dbatch_matmul.cc70 struct OpContext { struct
71 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
77 TfLiteBatchMatMulParams* params;
78 const TfLiteTensor* lhs;
79 const TfLiteTensor* rhs;
80 TfLiteTensor* output;
H A Dreduce.cc59 struct OpContext { struct
60 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
66 TfLiteReducerParams* params;
67 const TfLiteTensor* input;
68 const TfLiteTensor* axis;
69 TfLiteTensor* output;