Home
last modified time | relevance | path

Searched refs:TensorToGrid (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dserialization.cc238 data::TensorToGrid ToFB(TensorToGrid type) { in ToFB()
240 case TensorToGrid::kCustom: in ToFB()
241 return data::TensorToGrid::CUSTOM; in ToFB()
242 case TensorToGrid::kWBToX_HDToY_SToZ: in ToFB()
243 return data::TensorToGrid::WB_TO_X_HD_TO_Y_S_TO_Z; in ToFB()
244 case TensorToGrid::kWBToX_HDToY_ZIs1: in ToFB()
245 return data::TensorToGrid::WB_TO_X_HD_TO_Y_Z_IS_1; in ToFB()
246 case TensorToGrid::kWBToX_HToY_DToZ: in ToFB()
247 return data::TensorToGrid::WB_TO_X_H_TO_Y_D_TO_Z; in ToFB()
248 case TensorToGrid::kBToX_YIs1_ZIs1: in ToFB()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.h65 enum class TensorToGrid { enum
145 TensorToGrid tensor_to_grid_ = TensorToGrid::kCustom;
Dgpu_operation.cc217 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()
Dserialization_base.fbs200 enum TensorToGrid : byte {
232 tensor_to_grid:TensorToGrid;
Dserialization_base_generated.h337 enum class TensorToGrid : int8_t { enum
347 inline const TensorToGrid (&EnumValuesTensorToGrid())[5] { in EnumValuesTensorToGrid()
348 static const TensorToGrid values[] = { in EnumValuesTensorToGrid()
349 TensorToGrid::CUSTOM, TensorToGrid::WB_TO_X_HD_TO_Y_S_TO_Z, in EnumValuesTensorToGrid()
350 TensorToGrid::WB_TO_X_HD_TO_Y_Z_IS_1, TensorToGrid::WB_TO_X_H_TO_Y_D_TO_Z, in EnumValuesTensorToGrid()
351 TensorToGrid::B_TO_X_Y_IS_1_Z_IS_1}; in EnumValuesTensorToGrid()
365 inline const char *EnumNameTensorToGrid(TensorToGrid e) { in EnumNameTensorToGrid()
366 if (flatbuffers::IsOutRange(e, TensorToGrid::CUSTOM, in EnumNameTensorToGrid()
367 TensorToGrid::B_TO_X_Y_IS_1_Z_IS_1)) in EnumNameTensorToGrid()
2144 tflite::gpu::data::TensorToGrid tensor_to_grid() const { in tensor_to_grid()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dmax_unpooling.cc138 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateMaxUnpooling()
155 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateMaxUnpooling()
Dreshapex4.cc73 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateReshapex4()
Dsoftmax.cc79 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_ZIs1; in CreateSoftmax()
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()
Dspace_to_depth.cc80 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateSpaceToDepth()
Dreshape.cc92 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateReshape()
Dlstm.cc96 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateLSTM()
Dtranspose.cc105 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateTranspose()
Dpadding.cc142 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreatePadding()
Dpooling.cc305 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreatePooling()
329 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreatePooling()
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.cc280 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_ZIs1; in CreateConvConstants()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Ddepthwise_conv_plus_1x1_conv.cc261 result.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_ZIs1; in CreateDepthwiseConvPlus1x1Conv()