Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.cc97 tensor_to_grid_(operation.tensor_to_grid_), in GPUOperation()
119 tensor_to_grid_ = operation.tensor_to_grid_; in operator =()
217 if (elementwise_ || tensor_to_grid_ == TensorToGrid::kWBToX_HDToY_SToZ) { in GetGridSize()
223 if (tensor_to_grid_ == TensorToGrid::kWBToX_HDToY_ZIs1) { in GetGridSize()
229 if (tensor_to_grid_ == TensorToGrid::kWBToX_HToY_DToZ) { in GetGridSize()
235 if (tensor_to_grid_ == TensorToGrid::kBToX_YIs1_ZIs1) { in GetGridSize()
Dgpu_operation.h151 TensorToGrid tensor_to_grid_ = TensorToGrid::kCustom; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dspace_to_depth.cc124 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateSpaceToDepth()
135 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateDepthToSpace()
Dmax_unpooling.cc138 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateMaxUnpooling()
155 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateMaxUnpooling()
Ddepthwise_conv.cc260 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateDepthwiseConvolution2D()
286 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateDepthwiseConvolution2DDynamicWeights()
328 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateDepthwiseConvolution3D()
Dgather.cc60 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateGather()
Dreshapex4.cc73 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateReshapex4()
Dsoftmax.cc79 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_ZIs1; in CreateSoftmax()
Dresampler.cc82 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateResampler()
Dreshape.cc91 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateReshape()
Dlstm.cc96 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateLSTM()
Dtile.cc105 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateTile()
Dtranspose.cc104 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateTranspose()
Dpooling.cc305 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreatePooling()
329 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreatePooling()
Dpadding.cc141 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreatePadding()
Dconcat_xy.cc123 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateConcatXY()
Dconcat_z.cc152 op.tensor_to_grid_ = TensorToGrid::kWBToX_HToY_DToZ; in CreateConcatZ()
Dconv_constants.cc290 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_ZIs1; in CreateConvConstants()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Ddepthwise_conv_plus_1x1_conv.cc287 result.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_ZIs1; in CreateDepthwiseConvPlus1x1Conv()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dserialization.cc814 op->tensor_to_grid_ = ToEnum(fb_op->tensor_to_grid()); in Decode()
867 op_builder.add_tensor_to_grid(ToFB(op.tensor_to_grid_)); in Encode()