Searched defs:OpContext (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_context.h | 27 struct OpContext { struct
33 OpContext() { function_library = nullptr; } in OpContext() argument
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | maximum_minimum.cc | 42 struct OpContext { struct
43 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
48 const TfLiteEvalTensor* input1;
49 const TfLiteEvalTensor* input2;
50 TfLiteEvalTensor* output;
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fake_quant.cc | 33 struct OpContext { struct
34 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
38 const TfLiteTensor* input;
39 TfLiteTensor* output;
|
| D | split_v.cc | 33 struct OpContext { struct
34 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
40 TfLiteSplitVParams* params;
41 const TfLiteTensor* input;
42 const TfLiteTensor* size_splits;
43 const TfLiteTensor* axis;
|
| D | densify.cc | 30 struct OpContext { struct
31 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
35 const TfLiteTensor* input;
36 TfLiteTensor* output;
|
| D | split.cc | 31 struct OpContext { struct
32 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
37 TfLiteSplitParams* params;
38 const TfLiteTensor* axis;
39 const TfLiteTensor* input;
|
| D | dequantize.cc | 28 struct OpContext { struct
29 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
33 const TfLiteTensor* input;
34 TfLiteTensor* output;
|
| D | maximum_minimum.cc | 44 struct OpContext { struct
45 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
50 const TfLiteTensor* input1;
51 const TfLiteTensor* input2;
52 TfLiteTensor* output;
|
| D | numeric_verify.cc | 45 struct OpContext { struct
46 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
51 const TfLiteTensor* input;
52 const TfLiteTensor* ref;
53 TfLiteTensor* output;
|
| D | reduce.cc | 56 struct OpContext { struct
57 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
63 TfLiteReducerParams* params;
64 const TfLiteTensor* input;
65 const TfLiteTensor* axis;
66 TfLiteTensor* output;
|
| D | batch_matmul.cc | 70 struct OpContext { struct
71 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
77 TfLiteBatchMatMulParams* params;
78 const TfLiteTensor* lhs;
79 const TfLiteTensor* rhs;
80 TfLiteTensor* output;
|