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.cc87 OpContext op_context; in DescribeMatMul() local
124 OpContext op_context; in DescribeSparseTensorDenseMatMul() local
140 OpContext op_context; in DescribeXlaEinsum() local
157 OpContext op_context = DescribeXlaEinsum(dims_a, dims_b, equation); in DescribeEinsum() local
204 OpContext op_context; in DescribeConvolution() local
221 OpContext op_context; in DescribeDepthwiseConv2dNative() local
248 OpContext op_context; in DescribeFusedConv2DBiasActivation() local
304 OpContext op_context; in DescribeUnaryOp() local
322 OpContext op_context; in DescribeBinaryOp() local
338 OpContext op_context; in DescribeBiasAdd() local
[all …]
Dop_level_cost_estimator.cc364 return [this, impl](const OpContext& op_context, NodeCosts* node_costs) { in OpLevelCostEstimator()
694 Status OpLevelCostEstimator::PredictNodeCosts(const OpContext& op_context, in PredictNodeCosts()
772 Status OpLevelCostEstimator::PredictCwiseOp(const OpContext& op_context, in PredictCwiseOp()
805 const OpContext& op_context, NodeCosts* node_costs) const { in PredictCostOfAnUnknownOp() argument
1629 const int64 num_compute_ops, const OpContext& op_context, in PredictDefaultNodeCosts()
1660 Status OpLevelCostEstimator::PredictConv2D(const OpContext& op_context, in PredictConv2D()
1675 const OpContext& op_context, NodeCosts* node_costs) const { in PredictConv2DBackpropInput() argument
1691 const OpContext& op_context, NodeCosts* node_costs) const { in PredictConv2DBackpropFilter() argument
1707 const OpContext& op_context, NodeCosts* node_costs) const { in PredictFusedConv2DBiasActivation() argument
1792 Status OpLevelCostEstimator::PredictMatMul(const OpContext& op_context, in PredictMatMul()
[all …]
Danalytical_cost_estimator.cc39 void AddCostNode(ReadyNodeManager* node_manager, const OpContext& op_context, in AddCostNode()
194 OpContext op_context = scheduler_->GetCurrNode(); in PredictCosts() local
/external/tensorflow/tensorflow/lite/kernels/
Dreduce.cc82 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() local
260 OpContext op_context(context, node); in PrepareMeanOrSum() local
301 const OpContext& op_context, int num_axis, in EvalMeanReferenceOps()
369 OpContext op_context(context, node); in EvalMean() local
527 OpContext* op_context, T init_value, in EvalLogic()
579 OpContext* op_context, ReduceType reduce_type) { in EvalType()
[all …]
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()
98 void OneHotCompute(const OneHotContext& op_context) { in OneHotCompute()
107 const OneHotContext& op_context) { in ResizeOutputTensor()
126 OneHotContext op_context{context, node}; in Prepare() local
164 OneHotContext op_context{context, node}; in Eval() local
Dstrided_slice.cc69 StridedSliceParams BuildStridedSliceParams(StridedSliceContext* op_context) { in BuildStridedSliceParams()
103 StridedSliceContext* op_context) { in ResizeOutputTensor()
150 StridedSliceContext op_context(context, node); in Prepare() local
187 StridedSliceContext op_context(context, node); in Eval() local
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
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
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
Dbatch_to_space_nd.cc57 BatchToSpaceNDContext* op_context) { in ResizeOutputTensor()
98 BatchToSpaceNDContext op_context(context, node); in Prepare() local
115 BatchToSpaceNDContext op_context(context, node); in Eval() local
Dspace_to_batch_nd.cc58 SpaceToBatchNDContext* op_context) { in ResizeOutputTensor()
98 SpaceToBatchNDContext op_context(context, node); in Prepare() local
116 SpaceToBatchNDContext 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
Dexp.cc44 ExpContext op_context(context, node); in Prepare() local
52 ExpContext op_context(context, node); in Eval() local
Ddensify.cc57 OpContext op_context(context, node); in Prepare() local
72 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.cc81 OpContext op_context(context, node); in Prepare() local
107 OpContext op_context(context, node); in Eval() local
Dsplit_v.cc129 OpContext op_context(context, node); in Prepare() local
160 OpContext op_context(context, node); in Eval() local
/external/tensorflow/tensorflow/lite/micro/kernels/
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
Dstrided_slice.cc62 StridedSliceContext* op_context) { in BuildStridedSliceParams()
86 StridedSliceContext* op_context) { in CheckOutputSize()
131 StridedSliceContext op_context(context, node); in Prepare() local
Dmaximum_minimum.cc71 const OpContext& op_context) { in TFLiteOperation()
84 OpContext op_context(context, node); in Eval() local
/external/tensorflow/tensorflow/core/profiler/utils/
Dcost_utils.cc109 grappler::OpContext op_context; in Predict() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc171 OpKernelContext op_context(&params, n->num_outputs()); in Compile() local
227 OpKernelContext* op_context) { in CompileFunctionalNode()

12