Searched refs:weights_desc_ (Results 1 – 2 of 2) sorted by relevance
29 : GPUOperation(definition), weights_desc_(weights_desc) { in ConverterToConvWeights()30 code_ = GetConverterToConvWeightsCode(definition_, weights_desc_); in ConverterToConvWeights()36 weights_desc_(std::move(operation.weights_desc_)) {} in ConverterToConvWeights()41 weights_desc_ = std::move(operation.weights_desc_); in operator =()179 const int out_group_size = weights_desc_.GetOutputGroupSize(); in GetGridSize()
44 WeightsDescription weights_desc_; variable