Searched defs:partition_ids_t (Results 1 – 3 of 3) sorted by relevance
136 Tensor* partition_ids_t = nullptr; in SerializeScalarAccumulatorToOutput() local177 Tensor* partition_ids_t = nullptr; in SerializeTensorAccumulatorToOutput() local224 const Tensor& partition_ids_t, const Tensor& feature_ids_t, in AddToScalarAccumulator()253 const Tensor* partition_ids_t; in AddToScalarAccumulator() local267 const Tensor& partition_ids_t, const Tensor& feature_ids_t, in AddToTensorAccumulator()330 const Tensor* partition_ids_t; in AddToTensorAccumulator() local
150 const Tensor* partition_ids_t; in Compute() local437 const Tensor* partition_ids_t; in Compute() local712 const Tensor* partition_ids_t; in Compute() local
449 Tensor* partition_ids_t = nullptr; in DoCompute() local