Searched refs:feature_ids_t (Results 1 – 3 of 3) sorted by relevance
143 Tensor* feature_ids_t = nullptr; in SerializeScalarAccumulatorToOutput() local146 &feature_ids_t)); in SerializeScalarAccumulatorToOutput()147 auto feature_ids = feature_ids_t->matrix<int64>(); in SerializeScalarAccumulatorToOutput()183 Tensor* feature_ids_t = nullptr; in SerializeTensorAccumulatorToOutput() local186 &feature_ids_t)); in SerializeTensorAccumulatorToOutput()187 auto feature_ids = feature_ids_t->matrix<int64>(); in SerializeTensorAccumulatorToOutput()224 const Tensor& partition_ids_t, const Tensor& feature_ids_t, in AddToScalarAccumulator() argument230 const auto& feature_ids_and_dimensions = feature_ids_t.matrix<int64>(); in AddToScalarAccumulator()255 const Tensor* feature_ids_t; in AddToScalarAccumulator() local256 OP_REQUIRES_OK(context, context->input("feature_ids", &feature_ids_t)); in AddToScalarAccumulator()[all …]
716 const Tensor* feature_ids_t; in Compute() local717 OP_REQUIRES_OK(context, context->input("feature_ids", &feature_ids_t)); in Compute()718 const auto& feature_ids = feature_ids_t->matrix<int64>(); in Compute()
80 const Tensor* feature_ids_t; in Compute() local81 OP_REQUIRES_OK(context, context->input("feature_ids", &feature_ids_t)); in Compute()82 const auto feature_ids = feature_ids_t->vec<int32>(); in Compute()