Searched refs:weights_ (Results 1 – 2 of 2) sorted by relevance
28 weights_ = GetInput(operation, operands, kWeightsTensor); in RNN()84 const uint32_t num_units = weights_->shape().dimensions[0]; in Eval()86 const uint32_t input_weights_stride = weights_->shape().dimensions[1]; in Eval()103 const float* input_weights_ptr = reinterpret_cast<float*>(weights_->buffer); in Eval()
63 const RunTimeOperandInfo *weights_; variable