| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.h | 37 virtual Costs PredictCosts(const OpContext& op_context) const;
44 Costs PredictCostOfAnUnknownOp(const OpContext& op_context) const;
129 Costs PredictConv2D(const OpContext& op_context) const;
130 Costs PredictCwiseOp(const OpContext& op_context) const;
131 Costs PredictConv2DBackpropInput(const OpContext& op_context) const;
132 Costs PredictConv2DBackpropFilter(const OpContext& op_context) const;
133 Costs PredictFusedConv2DBiasActivation(const OpContext& op_context) const;
134 Costs PredictMatMul(const OpContext& op_context) const;
135 Costs PredictSparseTensorDenseMatMul(const OpContext& op_context) const;
136 Costs PredictNoOp(const OpContext& op_context) const;
[all …]
|
| D | op_level_cost_estimator_test.cc | 76 OpContext DescribeMatMul(int m, int n, int l, int k) { in DescribeMatMul()
77 OpContext op_context; in DescribeMatMul()
111 OpContext DescribeBatchMatMul(const std::vector<int>& dims_a, in DescribeBatchMatMul()
113 OpContext op_context; in DescribeBatchMatMul()
123 OpContext DescribeSparseTensorDenseMatMul(const int nnz_a, in DescribeSparseTensorDenseMatMul()
126 OpContext op_context; in DescribeSparseTensorDenseMatMul()
161 OpContext DescribeConvolution(int batch, int ix, int iy, int iz1, int iz2, in DescribeConvolution()
163 OpContext op_context; in DescribeConvolution()
178 OpContext DescribeDepthwiseConv2dNative(int batch, int ix, int iy, int iz1, in DescribeDepthwiseConv2dNative()
180 OpContext op_context; in DescribeDepthwiseConv2dNative()
[all …]
|
| D | op_level_cost_estimator.cc | 221 typedef Costs (OpLevelCostEstimator::*CostImpl)(const OpContext& op_context) in OpLevelCostEstimator()
223 auto wrap = [this](CostImpl impl) -> std::function<Costs(const OpContext&)> { in OpLevelCostEstimator()
224 return [this, impl](const OpContext& op_context) { in OpLevelCostEstimator()
373 Costs OpLevelCostEstimator::PredictCosts(const OpContext& op_context) const { in PredictCosts()
377 std::function<Costs(const OpContext&)> estimator = it->second; in PredictCosts()
448 Costs OpLevelCostEstimator::PredictCwiseOp(const OpContext& op_context) const { in PredictCwiseOp()
489 const OpContext& op_context) const { in PredictCostOfAnUnknownOp()
1100 Costs OpLevelCostEstimator::PredictConv2D(const OpContext& op_context) const { in PredictConv2D()
1111 const OpContext& op_context) const { in PredictConv2DBackpropInput()
1124 const OpContext& op_context) const { in PredictConv2DBackpropFilter()
[all …]
|
| D | op_context.h | 27 struct OpContext { struct
33 OpContext() { function_library = nullptr; } in OpContext() argument
|
| D | analytical_cost_estimator.cc | 39 void AddCostNode(ReadyNodeManager* node_manager, const OpContext& op_context, in AddCostNode()
166 OpContext op_context = scheduler_->GetCurrNode(); in PredictCosts()
|
| D | virtual_scheduler.cc | 618 OpContext VirtualScheduler::GetCurrNode() const { in GetCurrNode()
631 OpContext op_context; in GetCurrNode()
765 OpContext op_context = GetCurrNode(); in MarkCurrNodeExecuted()
|
| D | virtual_scheduler.h | 277 OpContext GetCurrNode() const;
|
| D | virtual_scheduler_test.cc | 1598 Costs SimplePredictCosts(const OpContext& op_context) const { in SimplePredictCosts()
1614 std::unordered_map<string, OpContext> RunScheduler( in RunScheduler()
1616 std::unordered_map<string, OpContext> ops_executed; in RunScheduler()
1619 OpContext op_context = scheduler_->GetCurrNode(); in RunScheduler()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fake_quant.cc | 34 struct OpContext { struct
35 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
58 OpContext op_context(context, node); in Prepare()
66 OpContext op_context(context, node); in Eval()
|
| D | maximum_minimum.cc | 38 struct OpContext { struct
39 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
53 OpContext op_context(context, node); in Prepare()
88 const OpContext& op_context) { in TFLiteOperation()
101 OpContext op_context(context, node); in Eval()
|
| D | reduce.cc | 48 struct OpContext { struct
49 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() argument
76 TfLiteStatus ResizeTempAxis(TfLiteContext* context, OpContext* op_context, in ResizeTempAxis()
84 TfLiteStatus ResizeTempSum(TfLiteContext* context, OpContext* op_context, in ResizeTempSum()
92 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, OpContext* op_context) { in ResizeOutputTensor()
161 OpContext* op_context) { in InitializeTemporaries()
211 OpContext op_context(context, node); in PrepareSimple()
240 OpContext op_context(context, node); in PrepareMeanOrSum()
271 OpContext op_context(context, node); in EvalMean()
402 OpContext* op_context, T init_value, in EvalLogic()
[all …]
|
| D | dequantize.cc | 31 struct OpContext { struct
32 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
59 OpContext op_context(context, node); in Prepare()
76 OpContext op_context(context, node); in Eval()
|
| D | split.cc | 30 struct OpContext { struct
31 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
74 OpContext op_context(context, node); in Prepare()
98 OpContext op_context(context, node); in Eval()
|
| D | split_v.cc | 29 struct OpContext { struct
30 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext() function
122 OpContext op_context(context, node); in Prepare()
150 OpContext op_context(context, node); in Eval()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule.cc | 33 OpContext op_context; in PredictExecutionTime()
|