Searched refs:splits_dtype (Results 1 – 3 of 3) sorted by relevance
79 Ragged(StringPiece feature_name, DataType dtype, DataType splits_dtype) in Ragged()83 splits_dtype(splits_dtype) {} in Ragged()88 DataType splits_dtype; member
962 if (!(c.splits_dtype == DT_INT32 || c.splits_dtype == DT_INT64)) { in CheckConfigDataTypes()964 DataTypeString(c.splits_dtype)); in CheckConfigDataTypes()1313 result->ragged_splits.emplace_back(config.ragged[d].splits_dtype, in FastParseExample()1316 if (config.ragged[d].splits_dtype == DT_INT64) { in FastParseExample()1335 if (config.ragged[d].splits_dtype == DT_INT64) { in FastParseExample()1508 const auto splits_dtype = config.ragged[d].splits_dtype; in FastParseSingleExample() local1509 result->ragged_splits.emplace_back(splits_dtype, TensorShape({0})); in FastParseSingleExample()2413 DataType splits_dtype = c.splits_dtype; in ParseContextRaggedFeatures() local2421 Tensor(allocator, splits_dtype, splits_shape); in ParseContextRaggedFeatures()2425 is_batch && splits_dtype == DT_INT32 in ParseContextRaggedFeatures()[all …]
46 const auto splits_dtype = DataTypeToEnum<SPLIT_TYPE>::v(); in BuildEncodeRaggedTensorGraph() local50 .Input(FakeInput(num_splits, splits_dtype)) // ragged_splits in BuildEncodeRaggedTensorGraph()54 .Attr("Tsplits", splits_dtype) in BuildEncodeRaggedTensorGraph()