Home
last modified time | relevance | path

Searched refs:kWBToX_HDToY_SToZ (Results 1 – 13 of 13) sorted by relevance

/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()
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()
Dreshapex4.cc73 op.tensor_to_grid_ = TensorToGrid::kWBToX_HDToY_SToZ; in CreateReshapex4()
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()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.h67 kWBToX_HDToY_SToZ, enumerator
Dgpu_operation.cc217 if (elementwise_ || tensor_to_grid_ == TensorToGrid::kWBToX_HDToY_SToZ) { in GetGridSize()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dserialization.cc242 case TensorToGrid::kWBToX_HDToY_SToZ: in ToFB()
286 return TensorToGrid::kWBToX_HDToY_SToZ; in ToEnum()