Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_context.h27 struct OpContext { struct
33 OpContext() { function_library = nullptr; } in OpContext() function
/external/tensorflow/tensorflow/lite/kernels/
Dfake_quant.cc34 struct OpContext { struct
35 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
39 const TfLiteTensor* input;
40 TfLiteTensor* output;
Dsplit_v.cc29 struct OpContext { struct
30 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
36 TfLiteSplitVParams* params;
37 const TfLiteTensor* input;
38 const TfLiteTensor* size_splits;
39 const TfLiteTensor* axis;
Dsplit.cc30 struct OpContext { struct
31 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
36 TfLiteSplitParams* params;
37 const TfLiteTensor* axis;
38 const TfLiteTensor* input;
Dmaximum_minimum.cc38 struct OpContext { struct
39 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
44 const TfLiteTensor* input1;
45 const TfLiteTensor* input2;
46 TfLiteTensor* output;
Ddequantize.cc31 struct OpContext { struct
32 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
36 const TfLiteTensor* input;
37 TfLiteTensor* output;
Dreduce.cc48 struct OpContext { struct
49 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
55 TfLiteReducerParams* params;
56 const TfLiteTensor* input;
57 const TfLiteTensor* axis;
58 TfLiteTensor* output;