Searched refs:loaded_tensor_t (Results 1 – 1 of 1) sorted by relevance
201 Tensor loaded_tensor_t; in Compute() local209 if (loaded_tensor_t.shape() != slice_shape) { in Compute()210 loaded_tensor_t = Tensor(DT_FLOAT, slice_shape); in Compute()213 &loaded_tensor_t)); in Compute()216 for (int row = 0; row < loaded_tensor_t.dim_size(0); ++row, ++row_index) { in Compute()233 const auto& loaded_tensor = loaded_tensor_t.matrix<float>(); in Compute()234 for (int old_col = 0; old_col < loaded_tensor_t.dim_size(1); in Compute()