Searched defs:sizes_tensor (Results 1 – 4 of 4) sorted by relevance
739 Tensor* sizes_tensor = nullptr; in Compute() local835 Tensor* sizes_tensor, std::vector<int32>* max_sizes) { in CountFields()
525 const Tensor* sizes_tensor; in Compute() local
416 const auto& sizes_tensor = context->tensors[node->inputs->data[2]]; in IsNodeSupportedByHexagon() local
2709 auto sizes_tensor = torch::empty( in __anonbfe5918f9502() local