Searched refs:out_ids (Results 1 – 2 of 2) sorted by relevance
242 Tensor* out_ids = nullptr; in Compute() local245 1, TensorShape({static_cast<int>(ids.size())}), &out_ids)); in Compute()254 out_ids->vec<int64>()(i) = ids[i]; in Compute()
923 std::vector<int32_t> out_ids(node.outputs.size()); in Encode() local924 for (int i = 0; i < out_ids.size(); ++i) { in Encode()925 out_ids[i] = node.outputs[i]; in Encode()928 auto out_ids_fb = builder->CreateVector(out_ids); in Encode()960 std::vector<int32_t> out_ids(inference.output_ids_.size()); in Encode() local961 for (int i = 0; i < out_ids.size(); ++i) { in Encode()962 out_ids[i] = inference.output_ids_[i]; in Encode()965 auto out_ids_fb = builder->CreateVector(out_ids); in Encode()