Searched refs:tfl_shape (Results 1 – 1 of 1) sorted by relevance
144 RuntimeShape tfl_shape(4, weights_->dims->data); in TransposeKernelWeights() local173 optimized_ops::Transpose<float>(params, tfl_shape, weights_->data.f, in TransposeKernelWeights()184 params, tfl_shape, reinterpret_cast<uint16_t*>(weights_->data.raw), in TransposeKernelWeights()