| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.h | 96 virtual Costs PredictCosts(const OpContext& op_context) const;
117 Status PredictNodeCosts(const OpContext& op_context,
121 Status PredictCostOfAnUnknownOp(const OpContext& op_context,
134 Status PredictNaryOp(const OpContext& op_context,
136 Status PredictConv2D(const OpContext& op_context,
138 Status PredictCwiseOp(const OpContext& op_context,
140 Status PredictConv2DBackpropInput(const OpContext& op_context,
142 Status PredictConv2DBackpropFilter(const OpContext& op_context,
144 Status PredictFusedConv2DBiasActivation(const OpContext& op_context,
146 Status PredictMatMul(const OpContext& op_context,
[all …]
|
| D | op_level_cost_estimator_test.cc | 86 OpContext DescribeMatMul(int m, int n, int l, int k) { in DescribeMatMul()
87 OpContext op_context; in DescribeMatMul()
121 OpContext DescribeSparseTensorDenseMatMul(const int nnz_a, in DescribeSparseTensorDenseMatMul()
124 OpContext op_context; in DescribeSparseTensorDenseMatMul()
137 OpContext DescribeXlaEinsum(const std::vector<int>& dims_a, in DescribeXlaEinsum()
140 OpContext op_context; in DescribeXlaEinsum()
154 OpContext DescribeEinsum(const std::vector<int>& dims_a, in DescribeEinsum()
157 OpContext op_context = DescribeXlaEinsum(dims_a, dims_b, equation); in DescribeEinsum()
202 OpContext DescribeConvolution(int batch, int ix, int iy, int iz1, int iz2, in DescribeConvolution()
204 OpContext op_context; in DescribeConvolution()
[all …]
|
| D | op_level_cost_estimator.cc | 251 bool IsEinsumCorrectlyFormed(const OpContext& einsum_context) { in IsEinsumCorrectlyFormed()
360 typedef Status (OpLevelCostEstimator::*CostImpl)(const OpContext& op_context, in OpLevelCostEstimator()
363 -> std::function<Status(const OpContext&, NodeCosts*)> { in OpLevelCostEstimator()
364 return [this, impl](const OpContext& op_context, NodeCosts* node_costs) { in OpLevelCostEstimator()
648 Costs OpLevelCostEstimator::PredictCosts(const OpContext& op_context) const { in PredictCosts()
694 Status OpLevelCostEstimator::PredictNodeCosts(const OpContext& op_context, in PredictNodeCosts()
699 std::function<Status(const OpContext&, NodeCosts*)> estimator = it->second; in PredictNodeCosts()
772 Status OpLevelCostEstimator::PredictCwiseOp(const OpContext& op_context, in PredictCwiseOp()
805 const OpContext& op_context, NodeCosts* node_costs) const { in PredictCostOfAnUnknownOp()
1111 const OpContext& einsum_context, OpContext* batch_matmul_context, in GenerateBatchMatmulContextFromEinsum()
[all …]
|
| D | op_context.h | 27 struct OpContext { struct
33 OpContext() { function_library = nullptr; } in OpContext() argument
|
| D | virtual_scheduler.h | 372 OpContext CreateOpContext(const NodeDef* node) const;
375 const OpContext& op_context);
479 virtual OpContext GetCurrNode();
|
| D | analytical_cost_estimator.cc | 39 void AddCostNode(ReadyNodeManager* node_manager, const OpContext& op_context, in AddCostNode()
194 OpContext op_context = scheduler_->GetCurrNode(); in PredictCosts()
|
| D | virtual_scheduler.cc | 745 OpContext SchedulerState::CreateOpContext(const NodeDef* node) const { in CreateOpContext()
756 OpContext op_context; in CreateOpContext()
868 const NodeDef* node, const Costs& node_costs, const OpContext& op_context) { in MarkNodeExecuted()
1347 OpContext VirtualScheduler::GetCurrNode() { in GetCurrNode()
|
| D | virtual_scheduler_test.cc | 2261 Costs SimplePredictCosts(const OpContext& op_context) const { in SimplePredictCosts()
2277 std::unordered_map<string, OpContext> RunScheduler( in RunScheduler()
2279 std::unordered_map<string, OpContext> ops_executed; in RunScheduler()
2282 OpContext op_context = scheduler_->GetCurrNode(); in RunScheduler()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fake_quant.cc | 33 struct OpContext { struct
34 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
57 OpContext op_context(context, node); in Prepare()
65 OpContext op_context(context, node); in Eval()
|
| D | maximum_minimum.cc | 44 struct OpContext { struct
45 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
59 OpContext op_context(context, node); in Prepare()
95 const OpContext& op_context) { in TFLiteOperation()
109 TfLiteContext* context, TfLiteNode* node, const OpContext& op_context) { in TFLiteOperation()
134 TfLiteContext* context, TfLiteNode* node, const OpContext& op_context) { in TFLiteOperation()
158 OpContext op_context(context, node); in Eval()
|
| D | dequantize.cc | 28 struct OpContext { struct
29 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
56 OpContext op_context(context, node); in Prepare()
80 OpContext op_context(context, node); in Eval()
|
| D | densify.cc | 30 struct OpContext { struct
31 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
57 OpContext op_context(context, node); in Prepare()
72 OpContext op_context(context, node); in Eval()
|
| D | split.cc | 31 struct OpContext { struct
32 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
81 OpContext op_context(context, node); in Prepare()
107 OpContext op_context(context, node); in Eval()
|
| D | reduce.cc | 56 struct OpContext { struct
57 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
82 TfLiteStatus ResizeTempAxis(TfLiteContext* context, OpContext* op_context, in ResizeTempAxis()
90 TfLiteStatus ResizeTempSum(TfLiteContext* context, OpContext* op_context, in ResizeTempSum()
98 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, OpContext* op_context) { in ResizeOutputTensor()
167 OpContext* op_context) { in InitializeTemporaries()
222 OpContext op_context(context, node); in PrepareSimple()
260 OpContext op_context(context, node); in PrepareMeanOrSum()
301 const OpContext& op_context, int num_axis, in EvalMeanReferenceOps()
369 OpContext op_context(context, node); in EvalMean()
[all …]
|
| D | numeric_verify.cc | 45 struct OpContext { struct
46 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
91 OpContext op_context(context, node); in Prepare()
128 static int32_t GetQuantizedValue(const OpContext& op_context, int index) { in GetQuantizedValue()
144 OpContext op_context(context, node); in Eval()
|
| D | split_v.cc | 33 struct OpContext { struct
34 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
129 OpContext op_context(context, node); in Prepare()
160 OpContext op_context(context, node); in Eval()
|
| D | batch_matmul.cc | 70 struct OpContext { struct
71 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
127 OpContext* op_context) { in InitializeTemporaries()
295 OpContext op_context(context, node); in Prepare()
652 OpContext op_context(context, node); in Eval()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | maximum_minimum.cc | 42 struct OpContext { struct
43 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
71 const OpContext& op_context) { in TFLiteOperation()
84 OpContext op_context(context, node); in Eval()
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | cost_utils.cc | 109 grappler::OpContext op_context; in Predict()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule.cc | 35 OpContext op_context; in PredictExecutionTime()
|