Searched refs:input_sizes_ (Results 1 – 3 of 3) sorted by relevance
37 (void)input_sizes_.emplace_back(size); in InitKernel()50 if (input_sizes_[i] == 0) { in Launch()57 Tensor tensor(data_type, shape, inputs[i]->addr, input_sizes_[i] * sizeof(T)); in Launch()
44 std::vector<size_t> input_sizes_; variable
54 int *input_sizes_; member