Home
last modified time | relevance | path

Searched defs:output_dense (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/c10d/
H A DProcessGroupNCCLTest.cpp474 auto output_dense = output_tensor[i].to(input_dense.device()).to_dense(); in testSparseAllreduce() local
525 auto output_dense = output_tensor[i].to(input_dense.device()).to_dense(); in testSparseAllreduceLarge() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/
H A Dparse_example.cc124 SeededHasher* hasher, std::vector<TfLiteTensor*>* output_dense, in FastParseSerializedExample()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dexample_proto_fast_parsing.cc590 SeededHasher hasher, std::vector<Tensor>* output_dense, in FastParseSerializedExample()