Searched refs:max_sequence_lengths (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 382 max_sequence_lengths=None, argument
445 max_sequence_lengths=max_sequence_lengths,
462 max_sequence_lengths=None, argument
525 max_sequence_lengths=max_sequence_lengths,
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | feature_column.py | 166 max_sequence_lengths=None, argument
237 if not max_sequence_lengths:
238 max_sequence_lengths = [0] * len(categorical_columns)
239 if len(max_sequence_lengths) != len(categorical_columns):
243 len(max_sequence_lengths), len(categorical_columns)))
275 categorical_columns, max_sequence_lengths):
|
| D | feature_column_v2.py | 216 max_sequence_lengths=None, argument
325 if not max_sequence_lengths:
326 max_sequence_lengths = [0] * len(categorical_columns)
327 if len(max_sequence_lengths) != len(categorical_columns):
331 len(max_sequence_lengths), len(categorical_columns)))
392 categorical_columns, max_sequence_lengths):
|
| D | tpu_embedding_v2.py | 984 max_sequence_lengths = []
1001 max_sequence_lengths.append(feature.max_sequence_length)
1025 max_sequence_lengths=max_sequence_lengths)
1034 max_sequence_lengths=max_sequence_lengths)
|
| D | feature_column_v2_test.py | 414 max_sequence_lengths=[0, 2],
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 87 name: "max_sequence_lengths"
182 name: "max_sequence_lengths"
|
| D | EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 222 name: "max_sequence_lengths"
317 name: "max_sequence_lengths"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tpu_update_embedding_enqueue_op_inputs.mlir | 20 …on = "0", combiners = ["mean", "sum"], device_ordinal = -1 : i64, max_sequence_lengths = [0, 0, 0]…
48 …on = "0", combiners = ["mean", "sum"], device_ordinal = -1 : i64, max_sequence_lengths = [0, 0, 0]…
61 …on = "0", combiners = ["mean", "sum"], device_ordinal = -1 : i64, max_sequence_lengths = [0, 0, 0]…
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 87 name: "max_sequence_lengths"
|
| D | EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 222 name: "max_sequence_lengths"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.tpu.experimental.pbtxt | 45 …biner\', \'initializer\', \'shared_embedding_collection_name\', \'max_sequence_lengths\', \'learni…
|
| D | tensorflow.raw_ops.pbtxt | 1349 …e_override\', \'table_ids\', \'device_ordinal\', \'combiners\', \'max_sequence_lengths\', \'num_fe…
1357 …e_override\', \'table_ids\', \'device_ordinal\', \'combiners\', \'max_sequence_lengths\', \'num_fe…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1349 …e_override\', \'table_ids\', \'device_ordinal\', \'combiners\', \'max_sequence_lengths\', \'num_fe…
1357 …e_override\', \'table_ids\', \'device_ordinal\', \'combiners\', \'max_sequence_lengths\', \'num_fe…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 14096 name: "max_sequence_lengths"
14282 name: "max_sequence_lengths"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 3773 DefaultValuedAttr<I64ArrayAttr, "{}">:$max_sequence_lengths,
3861 DefaultValuedAttr<I64ArrayAttr, "{}">:$max_sequence_lengths,
|