Home
last modified time | relevance | path

Searched refs:tensor_to_grid_ (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dgpu_operation.cc139 tensor_to_grid_(operation.tensor_to_grid_), in GPUOperation()
164 tensor_to_grid_ = operation.tensor_to_grid_; in operator =()
462 if (tensor_to_grid_ == TensorToGrid::kWBToX_HDToY_SToZ) { in GetGridSize()
468 if (tensor_to_grid_ == TensorToGrid::kWBToX_HDToY_ZIs1) { in GetGridSize()
474 if (tensor_to_grid_ == TensorToGrid::kWBToX_HToY_DToZ) { in GetGridSize()
480 if (tensor_to_grid_ == TensorToGrid::kBToX_YIs1_ZIs1) { in GetGridSize()
541 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateGpuOperation()
590 result->tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in Fuse2InputElemWith2SimpleElem()
H A Dgpu_operation.h178 TensorToGrid tensor_to_grid_ = TensorToGrid::kCustom; variable
H A Dserialization_base.cc654 op->tensor_to_grid_ = ToEnum(fb_op->tensor_to_grid()); in Decode()
707 op_builder.add_tensor_to_grid(ToFB(op.tensor_to_grid_)); in Encode()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dspace_to_depth.cc120 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateSpaceToDepth()
131 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateDepthToSpace()
H A Delementwise.cc409 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateElementwiseOneInputWithBroadcast()
434 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateElementwiseWithBroadcast()
458 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateElementwiseTwoInputWithBroadcast()
H A Dmax_unpooling.cc131 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateMaxUnpooling()
149 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateMaxUnpooling()
H A Ddepthwise_conv.cc422 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateDepthwiseConvolution2D()
444 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateDepthwiseConvolution2DDynamicWeights()
480 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateDepthwiseConvolution3D()
H A Dpooling.cc267 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreatePooling()
289 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreatePooling()
H A Dgather.cc60 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateGather()
H A Dreshapex4.cc74 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateReshapex4()
H A Dsoftmax.cc79 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_ZIs1; in CreateSoftmax()
H A Dresampler.cc86 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateResampler()
H A Dreshape.cc88 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateReshape()
H A Done_hot.cc64 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateOneHot()
H A Dlstm.cc96 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateLSTM()
H A Dselect_v2.cc86 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateSelectV2()
H A Dtile.cc104 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateTile()
H A Dtranspose.cc101 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateTranspose()
H A Dadd.cc35 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateUnequalAdd()
H A Dpadding.cc142 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreatePadding()
H A Dconcat_xy.cc123 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateConcatXY()
H A Dconcat_z.cc152 op.tensor_to_grid_ = TensorToGrid::kWBToX_HToY_DToZ; in CreateConcatZ()
H A Dconv_constants.cc269 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_ZIs1; in CreateConvConstants()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
H A Dconv_pointwise.cc199 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateConvPointwise()
H A Ddepthwise_conv_plus_1x1_conv.cc591 operation.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_ZIs1; in Finalize()