Home
last modified time | relevance | path

Searched defs:op_context (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc88 OpContext op_context; in DescribeMatMul() local
125 OpContext op_context; in DescribeSparseTensorDenseMatMul() local
141 OpContext op_context; in DescribeXlaEinsum() local
158 OpContext op_context = DescribeXlaEinsum(dims_a, dims_b, equation); in DescribeEinsum() local
205 OpContext op_context; in DescribeConvolution() local
222 OpContext op_context; in DescribeDepthwiseConv2dNative() local
249 OpContext op_context; in DescribeFusedConv2DBiasActivation() local
305 OpContext op_context; in DescribeUnaryOp() local
323 OpContext op_context; in DescribeBinaryOp() local
339 OpContext op_context; in DescribeBiasAdd() local
[all …]
Dop_level_cost_estimator.cc365 return [this, impl](const OpContext& op_context, NodeCosts* node_costs) { in OpLevelCostEstimator()
697 Status OpLevelCostEstimator::PredictNodeCosts(const OpContext& op_context, in PredictNodeCosts()
784 Status OpLevelCostEstimator::PredictCwiseOp(const OpContext& op_context, in PredictCwiseOp()
817 const OpContext& op_context, NodeCosts* node_costs) const { in PredictCostOfAnUnknownOp() argument
1661 const int64_t num_compute_ops, const OpContext& op_context, in PredictDefaultNodeCosts()
1692 Status OpLevelCostEstimator::PredictConv2D(const OpContext& op_context, in PredictConv2D()
1708 const OpContext& op_context, NodeCosts* node_costs) const { in PredictConv2DBackpropInput() argument
1724 const OpContext& op_context, NodeCosts* node_costs) const { in PredictConv2DBackpropFilter() argument
1740 const OpContext& op_context, NodeCosts* node_costs) const { in PredictFusedConv2DBiasActivation() argument
1825 Status OpLevelCostEstimator::PredictMatMul(const OpContext& op_context, in PredictMatMul()
[all …]
Danalytical_cost_estimator.cc40 Status AddCostNode(ReadyNodeManager* node_manager, const OpContext& op_context, in AddCostNode()
200 OpContext op_context = scheduler_->GetCurrNode(); in PredictCosts() local
/external/tensorflow/tensorflow/lite/kernels/
Dmaximum_minimum.cc59 OpContext op_context(context, node); in Prepare() local
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() local
Done_hot.cc64 void OneHotComputeImpl(const OneHotContext& op_context) { in OneHotComputeImpl()
103 void OneHotCompute(const OneHotContext& op_context) { in OneHotCompute()
112 const OneHotContext& op_context) { in ResizeOutputTensor()
131 OneHotContext op_context{context, node}; in Prepare() local
169 OneHotContext op_context{context, node}; in Eval() local
Dreduce.cc85 TfLiteStatus ResizeTempAxis(TfLiteContext* context, OpContext* op_context, in ResizeTempAxis()
93 TfLiteStatus ResizeTempAccum(TfLiteContext* context, OpContext* op_context, in ResizeTempAccum()
101 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, OpContext* op_context) { in ResizeOutputTensor()
169 TfLiteStatus ResizeTempDims(TfLiteContext* context, OpContext* op_context, in ResizeTempDims()
178 OpContext* op_context) { in InitializeTemporaries()
240 OpContext op_context(context, node); in PrepareSimple() local
289 OpContext op_context(context, node); in PrepareMeanOrSum() local
329 OpContext op_context(context, node); in PrepareProd() local
377 TfLiteStatus Mean(TfLiteContext* context, const OpContext* op_context, in Mean()
397 const OpContext* op_context, int* temp_index, in QuantizedMeanOrSum()
[all …]
Dbroadcast_to.cc48 BroadcastToContext* op_context) { in ResizeOutputTensor()
96 BroadcastToContext op_context(context, node); in Prepare() local
113 BroadcastToContext op_context(context, node); in Eval() local
Dpad.cc76 PadContext* op_context) { in ResizeOutputTensor()
113 PadContext op_context(context, node); in Prepare() local
137 TfLiteStatus EvalInt(TfLiteContext* context, const PadContext& op_context, in EvalInt()
176 PadContext op_context(context, node); in Eval() local
Dtranspose.cc49 TransposeContext* op_context) { in ResizeOutputTensor()
75 TransposeContext op_context(context, node); in Prepare() local
92 TransposeContext op_context(context, node); in Eval() local
Dbatch_to_space_nd.cc57 BatchToSpaceNDContext* op_context) { in ResizeOutputTensor()
99 BatchToSpaceNDContext op_context(context, node); in Prepare() local
116 BatchToSpaceNDContext op_context(context, node); in Eval() local
Dspace_to_batch_nd.cc58 SpaceToBatchNDContext* op_context) { in ResizeOutputTensor()
99 SpaceToBatchNDContext op_context(context, node); in Prepare() local
117 SpaceToBatchNDContext op_context(context, node); in Eval() local
Dstrided_slice.cc72 StridedSliceParams BuildStridedSliceParams(StridedSliceContext* op_context) { in BuildStridedSliceParams()
187 StridedSliceContext* op_context) { in ResizeOutputTensor()
237 StridedSliceContext op_context(context, node); in Prepare() local
268 StridedSliceContext op_context(context, node); in Eval() local
Dexp.cc44 ExpContext op_context(context, node); in Prepare() local
52 ExpContext op_context(context, node); in Eval() local
Dfake_quant.cc57 OpContext op_context(context, node); in Prepare() local
65 OpContext op_context(context, node); in Eval() local
Dbroadcast_args.cc49 BroadcastArgsContext op_context(context, node); in Prepare() local
67 BroadcastArgsContext op_context(context, node); in Eval() local
Ddensify.cc57 OpContext op_context(context, node); in Prepare() local
73 OpContext op_context(context, node); in Eval() local
Ddequantize.cc56 OpContext op_context(context, node); in Prepare() local
80 OpContext op_context(context, node); in Eval() local
Dsqueeze.cc42 SqueezeContext op_context(context, node); in Prepare() local
81 SqueezeContext op_context(context, node); in Eval() local
Dnumeric_verify.cc91 OpContext op_context(context, node); in Prepare() local
128 static int32_t GetQuantizedValue(const OpContext& op_context, int index) { in GetQuantizedValue()
144 OpContext op_context(context, node); in Eval() local
Dsplit.cc82 OpContext op_context(context, node); in Prepare() local
108 OpContext op_context(context, node); in Eval() local
Dsplit_v.cc132 OpContext op_context(context, node); in Prepare() local
163 OpContext op_context(context, node); in Eval() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc172 OpKernelContext op_context(&params, n->num_outputs()); in Compile() local
228 OpKernelContext* op_context) { in CompileFunctionalNode()
/external/tensorflow/tensorflow/core/profiler/utils/
Dcost_utils.cc108 grappler::OpContext op_context; in Predict() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc91 OpKernelContext op_context(&params); in EvaluateNode() local
/external/tensorflow/tensorflow/core/transforms/utils/
Deval_utils.cc135 tensorflow::OpKernelContext op_context(&params); in EvaluateOperation() local

12