| /external/tensorflow/tensorflow/lite/kernels/ |
| D | one_hot.cc | 62 void OneHotComputeImpl(const OneHotContext& op_context) { in OneHotComputeImpl()
96 void OneHotCompute(const OneHotContext& op_context) { in OneHotCompute()
105 const OneHotContext& op_context) { in ResizeOutputTensor()
124 OneHotContext op_context{context, node}; in Prepare() local
159 OneHotContext op_context{context, node}; in Eval() local
|
| D | reduce.cc | 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() local
240 OpContext op_context(context, node); in PrepareMeanOrSum() local
271 OpContext op_context(context, node); in EvalMean() local
402 OpContext* op_context, T init_value, in EvalLogic()
442 OpContext* op_context, ReduceType reduce_type) { in EvalType()
476 OpContext* op_context, ReduceType reduce_type) { in EvalType()
[all …]
|
| D | strided_slice.cc | 81 inline int32_t GetBeginValueAtIndex(StridedSliceContext* op_context, int idx) { in GetBeginValueAtIndex()
90 inline int32_t GetEndValueAtIndex(StridedSliceContext* op_context, int idx) { in GetEndValueAtIndex()
103 StridedSliceContext* op_context) { in ResizeOutputTensor()
146 StridedSliceContext op_context(context, node); in Prepare() local
180 StridedSliceContext op_context(context, node); in Eval() local
|
| D | transpose.cc | 46 TransposeContext* op_context) { in ResizeOutputTensor()
72 TransposeContext op_context(context, node); in Prepare() local
88 TransposeContext op_context(context, node); in Eval() local
|
| D | maximum_minimum.cc | 53 OpContext op_context(context, node); in Prepare() local
88 const OpContext& op_context) { in TFLiteOperation()
101 OpContext op_context(context, node); in Eval() local
|
| D | space_to_batch_nd.cc | 57 SpaceToBatchNDContext* op_context) { in ResizeOutputTensor()
94 SpaceToBatchNDContext op_context(context, node); in Prepare() local
109 SpaceToBatchNDContext op_context(context, node); in Eval() local
|
| D | batch_to_space_nd.cc | 57 BatchToSpaceNDContext* op_context) { in ResizeOutputTensor()
105 BatchToSpaceNDContext op_context(context, node); in Prepare() local
120 BatchToSpaceNDContext op_context(context, node); in Eval() local
|
| D | exp.cc | 47 ExpContext op_context(context, node); in Prepare() local
55 ExpContext op_context(context, node); in Eval() local
|
| D | fake_quant.cc | 58 OpContext op_context(context, node); in Prepare() local
66 OpContext op_context(context, node); in Eval() local
|
| D | pad.cc | 71 PadContext* op_context) { in ResizeOutputTensor()
100 PadContext op_context(context, node); in Prepare() local
121 PadContext op_context(context, node); in Eval() local
|
| D | squeeze.cc | 42 SqueezeContext op_context(context, node); in Prepare() local
81 SqueezeContext op_context(context, node); in Eval() local
|
| D | dequantize.cc | 59 OpContext op_context(context, node); in Prepare() local
76 OpContext op_context(context, node); in Eval() local
|
| D | split.cc | 74 OpContext op_context(context, node); in Prepare() local
98 OpContext op_context(context, node); in Eval() local
|
| D | split_v.cc | 122 OpContext op_context(context, node); in Prepare() local
150 OpContext op_context(context, node); in Eval() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 77 OpContext op_context; in DescribeMatMul() local
113 OpContext op_context; in DescribeBatchMatMul() local
126 OpContext op_context; in DescribeSparseTensorDenseMatMul() local
163 OpContext op_context; in DescribeConvolution() local
180 OpContext op_context; in DescribeDepthwiseConv2dNative() local
203 OpContext op_context; in DescribeFusedConv2DBiasActivation() local
244 OpContext op_context; in DescribeUnaryOp() local
262 OpContext op_context; in DescribeBinaryOp() local
278 OpContext op_context; in DescribeBiasAdd() local
372 OpContext op_context; in DescribePoolingOp() local
[all …]
|
| 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() local
|
| D | virtual_scheduler.cc | 631 OpContext op_context; in GetCurrNode() local
765 OpContext op_context = GetCurrNode(); in MarkCurrNodeExecuted() local
|
| D | op_level_cost_estimator.cc | 224 return [this, impl](const OpContext& op_context) { in OpLevelCostEstimator()
1375 const OpContext& op_context, in PredictFusedOp()
|
| D | virtual_scheduler_test.cc | 1619 OpContext op_context = scheduler_->GetCurrNode(); in RunScheduler() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.cc | 164 OpKernelContext op_context(¶ms, n->num_outputs()); in Compile() local
220 OpKernelContext* op_context) { in CompileFunctionalNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 112 OpKernelContext op_context(¶ms); in EvaluateNode() local
|
| D | static_schedule.cc | 33 OpContext op_context; in PredictExecutionTime() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_matmul_op.cc | 35 void GemmlowpMultiply(OpKernelContext* op_context, const quint8* a_data, in GemmlowpMultiply()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 110 OpKernelContext* const op_context, in CreateIteratorContext()
|