Searched refs:TensorToGrid (Results 1 – 22 of 22) sorted by relevance
241 data::TensorToGrid ToFB(TensorToGrid type) { in ToFB()243 case TensorToGrid::kCustom: in ToFB()244 return data::TensorToGrid::CUSTOM; in ToFB()245 case TensorToGrid::kWBToX_HDToY_SToZ: in ToFB()246 return data::TensorToGrid::WB_TO_X_HD_TO_Y_S_TO_Z; in ToFB()247 case TensorToGrid::kWBToX_HDToY_ZIs1: in ToFB()248 return data::TensorToGrid::WB_TO_X_HD_TO_Y_Z_IS_1; in ToFB()249 case TensorToGrid::kWBToX_HToY_DToZ: in ToFB()250 return data::TensorToGrid::WB_TO_X_H_TO_Y_D_TO_Z; in ToFB()251 case TensorToGrid::kBToX_YIs1_ZIs1: in ToFB()[all …]
71 enum class TensorToGrid { enum151 TensorToGrid tensor_to_grid_ = TensorToGrid::kCustom;
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()
200 enum TensorToGrid : byte {232 tensor_to_grid:TensorToGrid;
337 enum class TensorToGrid : int8_t { enum347 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 …]
124 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateSpaceToDepth()135 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateDepthToSpace()
138 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateMaxUnpooling()155 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateMaxUnpooling()
260 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()
60 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateGather()
73 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateReshapex4()
79 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_ZIs1; in CreateSoftmax()
82 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateResampler()
91 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateReshape()
96 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateLSTM()
105 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateTile()
104 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateTranspose()
305 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreatePooling()329 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreatePooling()
141 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreatePadding()
123 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateConcatXY()
152 op.tensor_to_grid_ = TensorToGrid::kWBToX_HToY_DToZ; in CreateConcatZ()
290 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_ZIs1; in CreateConvConstants()
287 result.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_ZIs1; in CreateDepthwiseConvPlus1x1Conv()