Searched refs:max_sequence_lengths (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 395 max_sequence_lengths=None, argument
458 max_sequence_lengths=max_sequence_lengths,
475 max_sequence_lengths=None, argument
538 max_sequence_lengths=max_sequence_lengths,
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | feature_column.py | 162 max_sequence_lengths=None, argument
234 if not max_sequence_lengths:
235 max_sequence_lengths = [0] * len(categorical_columns)
236 if len(max_sequence_lengths) != len(categorical_columns):
240 len(max_sequence_lengths), len(categorical_columns)))
272 categorical_columns, max_sequence_lengths):
|
| D | feature_column_v2.py | 212 max_sequence_lengths=None, argument
322 if not max_sequence_lengths:
323 max_sequence_lengths = [0] * len(categorical_columns)
324 if len(max_sequence_lengths) != len(categorical_columns):
328 len(max_sequence_lengths), len(categorical_columns)))
389 categorical_columns, max_sequence_lengths):
|
| D | feature_column_v2_test.py | 432 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 | 49 …biner\', \'initializer\', \'shared_embedding_collection_name\', \'max_sequence_lengths\', \'learni…
|
| D | tensorflow.raw_ops.pbtxt | 1429 …e_override\', \'table_ids\', \'device_ordinal\', \'combiners\', \'max_sequence_lengths\', \'num_fe…
1437 …e_override\', \'table_ids\', \'device_ordinal\', \'combiners\', \'max_sequence_lengths\', \'num_fe…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1429 …e_override\', \'table_ids\', \'device_ordinal\', \'combiners\', \'max_sequence_lengths\', \'num_fe…
1437 …e_override\', \'table_ids\', \'device_ordinal\', \'combiners\', \'max_sequence_lengths\', \'num_fe…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 15745 name: "max_sequence_lengths"
15931 name: "max_sequence_lengths"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 4394 DefaultValuedAttr<I64ArrayAttr, "{}">:$max_sequence_lengths,
4482 DefaultValuedAttr<I64ArrayAttr, "{}">:$max_sequence_lengths,
|