Searched refs:loaded_tensor_t (Results 1 – 1 of 1) sorted by relevance
202 Tensor loaded_tensor_t; in Compute() local210 if (loaded_tensor_t.shape() != slice_shape) { in Compute()211 loaded_tensor_t = Tensor(DT_FLOAT, slice_shape); in Compute()214 &loaded_tensor_t)); in Compute()217 for (int row = 0; row < loaded_tensor_t.dim_size(0); ++row, ++row_index) { in Compute()234 const auto& loaded_tensor = loaded_tensor_t.matrix<float>(); in Compute()235 for (int old_col = 0; old_col < loaded_tensor_t.dim_size(1); in Compute()