Searched refs:tensor_to_grid_ (Results 1 – 20 of 20) sorted by relevance
97 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()
151 TensorToGrid tensor_to_grid_ = TensorToGrid::kCustom; variable
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()
814 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()