blob: 9ec45571159093d05b636d7cae60c611b35e852d [file] [log] [blame]
op {
name: "LogUniformCandidateSampler"
input_arg {
name: "true_classes"
type: DT_INT64
}
output_arg {
name: "sampled_candidates"
type: DT_INT64
}
output_arg {
name: "true_expected_count"
type: DT_FLOAT
}
output_arg {
name: "sampled_expected_count"
type: DT_FLOAT
}
attr {
name: "num_true"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "num_sampled"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "unique"
type: "bool"
}
attr {
name: "range_max"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "seed"
type: "int"
default_value {
i: 0
}
}
attr {
name: "seed2"
type: "int"
default_value {
i: 0
}
}
}
op {
name: "LogUniformCandidateSampler"
input_arg {
name: "true_classes"
type: DT_INT64
}
output_arg {
name: "sampled_candidates"
type: DT_INT64
}
output_arg {
name: "true_expected_count"
type: DT_FLOAT
}
output_arg {
name: "sampled_expected_count"
type: DT_FLOAT
}
attr {
name: "num_true"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "num_sampled"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "unique"
type: "bool"
}
attr {
name: "range_max"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "seed"
type: "int"
default_value {
i: 0
}
}
attr {
name: "seed2"
type: "int"
default_value {
i: 0
}
}
is_stateful: true
}