Searched defs:reshaped_weights (Results 1 – 3 of 3) sorted by relevance
343 …const ITensorInfo &reshaped_weights = TensorInfo(weights->clone()->set_is_resizable(true).reset_p… in validate() local459 …CLAuxTensorHandler reshaped_weights(offset_int_vec(TransposedWeights), _reshaped_weights, tensors,… in prepare() local
384 …const ITensorInfo &reshaped_weights = TensorInfo(weights->clone()->set_is_resizable(true).reset_p… in validate() local498 …CpuAuxTensorHandler reshaped_weights(offset_int_vec(TransposedWeights), _reshaped_weights, tensors… in prepare() local
270 auto reshaped_weights = xla::Reshape(weights, /*dimensions=*/reshape_dims, in CalculateGradData() local