Searched refs:scale_tf (Results 1 – 1 of 1) sorted by relevance
818 const U* scale_tf = scale_tensor.flat<U>().data(); in Compute() local821 std::memcpy(weights_data, scale_tf, depth_ * sizeof(U)); in Compute()1212 const U* scale_tf = scale_tensor.flat<U>().data(); in Compute() local1214 weights_data_tf[k] = scale_tf[k]; in Compute()