Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A 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 …]
H A 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 …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A 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 …]
H A 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
H A 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
H A 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
H A 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
H A 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
H A 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
H A 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
H A 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
H A 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
H A Dfake_quant.cc57 OpContext op_context(context, node); in Prepare() local
65 OpContext op_context(context, node); in Eval() local
H A Dexp.cc44 ExpContext op_context(context, node); in Prepare() local
52 ExpContext op_context(context, node); in Eval() local
H A Dbroadcast_args.cc49 BroadcastArgsContext op_context(context, node); in Prepare() local
67 BroadcastArgsContext op_context(context, node); in Eval() local
H A Ddensify.cc57 OpContext op_context(context, node); in Prepare() local
73 OpContext op_context(context, node); in Eval() local
H A Ddequantize.cc56 OpContext op_context(context, node); in Prepare() local
80 OpContext op_context(context, node); in Eval() local
H A Dsqueeze.cc42 SqueezeContext op_context(context, node); in Prepare() local
81 SqueezeContext op_context(context, node); in Eval() local
H A Dsplit.cc82 OpContext op_context(context, node); in Prepare() local
108 OpContext op_context(context, node); in Eval() local
H A Dsplit_v.cc132 OpContext op_context(context, node); in Prepare() local
163 OpContext op_context(context, node); in Eval() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/
H A DRegisterOpContextClass.cpp19 -> SerializationTypeLinearPrePack { // __getstate__ in TORCH_LIBRARY()
35 -> SerializationTypeConv2dPrePack { // __getstate__ in TORCH_LIBRARY()
55 -> SerializationTypeTransposeConv2dPrePack { // __getstate__ in TORCH_LIBRARY()
H A DConvolution.cpp99 auto op_context = create( in create_and_run() local
442 const c10::intrusive_ptr<xnnpack::Conv2dOpContext>& op_context) { in conv2d_clamp_run()
449 auto op_context = ivalue.toCustomClass<xnnpack::Conv2dOpContext>(); in unpack_prepacked_sizes_conv2d() local
463 const c10::intrusive_ptr<xnnpack::TransposeConv2dOpContext>& op_context) { in conv2d_transpose_clamp_run()
H A DOpContext.h66 ContextLinear&& op_context) in XNNPackLinearOpContext()
168 ContextConv2D&& op_context) in XNNPackConv2dOpContext()
217 ContextConv2D&& op_context) in XNNPackTransposeConv2dOpContext()
H A DOpContext.cpp56 auto op_context = in create_context() local
100 auto op_context = in create_context() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/
H A DMetalPrepackOpRegister.cpp46 -> SerializationTypeConv2dPrePack { // __getstate__ in TORCH_LIBRARY()
64 -> SerializationTypeLinearPrePack { // __getstate__ in TORCH_LIBRARY()

12