| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sdca_ops_test.cc | 88 const int num_examples) { in SparseExampleIndices()
96 const int num_examples) { in SparseFeatureIndices()
122 void GetGraphs(const int32_t num_examples, in GetGraphs()
232 const int num_examples = state.range(0); in BM_SDCA() local
244 const int num_examples = state.range(0); in BM_SDCA_LARGE_DENSE() local
257 const int num_examples = state.range(0); in BM_SDCA_LARGE_SPARSE() local
|
| H A D | sdca_internal.cc | 379 const int num_examples = static_cast<int>(example_weights.size()); in Initialize() local
421 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateSparseFeatureRepresentation()
508 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateDenseFeatureRepresentation()
540 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in ComputeSquaredNormPerExample()
|
| H A D | sdca_internal.h | 337 int num_examples() const { return examples_.size(); } in num_examples() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | parsing_ops.cc | 66 const DimensionHandle& num_examples, in AddRaggedOutputShapes()
214 DimensionHandle num_examples = c->UnknownDim(); in __anonf9b685fd0502() local
293 DimensionHandle num_examples = c->Dim(input, 0); in __anonf9b685fd0702() local
376 DimensionHandle num_examples = c->UnknownDim(); in __anonf9b685fd0802() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | example_proto_fast_parsing.cc | 2296 bool is_batch, int num_examples, in ParseContextDenseFeatures()
2359 bool is_batch, int num_examples, in ParseContextSparseFeatures()
2421 bool is_batch, int num_examples, in ParseContextRaggedFeatures()
2499 bool is_batch, int num_examples, in ParseSequenceDenseFeatures()
2653 gtl::ArraySlice<tstring> example_names, bool is_batch, int num_examples, in ParseSequenceSparseFeatures()
2781 gtl::ArraySlice<tstring> example_names, bool is_batch, int num_examples, in ParseSequenceRaggedFeatures()
2933 int num_examples = serialized.size(); in FastParseSequenceExample() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/gcs_test/python/ |
| H A D | gcs_smoke.py | 35 def create_examples(num_examples, input_mean): argument
|
| /aosp_15_r20/external/federated-compute/fcp/client/ |
| H A D | client_runner.h | 122 explicit FakeExampleIterator(int num_examples) in FakeExampleIterator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/ |
| H A D | quantize_model_test_base.py | 195 num_examples=8) -> repr_dataset.RepresentativeDataset: argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A D | auto_shard_dataset_test.py | 68 num_examples, shuffle): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | bidirectional_sequence_rnn_test.cc | 1129 const int num_examples = 64; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| H A D | parse_example.cc | 434 int num_examples, int batch_size, in CopyToBuffer()
|