Searched refs:sparse_int_feature_shapes_list (Results 1 – 5 of 5) sorted by relevance
207 OpInputList sparse_int_feature_shapes_list; in DoCompute() local211 &sparse_int_feature_shapes_list)); in DoCompute()216 sparse_int_feature_shapes_list); in DoCompute()234 sparse_int_feature_shapes_list))); in DoCompute()419 OpInputList sparse_int_feature_shapes_list; in DoCompute() local423 &sparse_int_feature_shapes_list)); in DoCompute()428 sparse_int_feature_shapes_list); in DoCompute()446 sparse_int_feature_shapes_list))); in DoCompute()
88 const OpInputList& sparse_int_feature_shapes_list) { in InferBatchSize() argument95 if (sparse_int_feature_shapes_list.size() > 0) { in InferBatchSize()96 return sparse_int_feature_shapes_list[0].flat<int64>()(0); in InferBatchSize()
32 std::vector<Tensor> sparse_int_feature_shapes_list) { in Initialize() argument111 sparse_int_feature_shapes_list.size() == num_sparse_int_features, in Initialize()120 sparse_int_feature_shapes_list[sparse_feat_idx]; in Initialize()
53 const OpInputList& sparse_int_feature_shapes_list);
46 std::vector<Tensor> sparse_int_feature_shapes_list);