Home
last modified time | relevance | path

Searched defs:tensor_ids (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/
H A DGpuKernelComponentGraph.cpp36 std::vector<DependencyGraph::TensorId> tensor_ids{}; in get_tensor_ids() local
H A DGpuOperatorGroup.cpp38 std::vector<DependencyGraph::TensorId> tensor_ids{}; in get_tensor_ids() local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dtape.h386 gtl::ArraySlice<int64_t> tensor_ids, in ShouldRecord()
901 gtl::ArraySlice<int64_t> tensor_ids, in ShouldRecord()
H A Dgradients.cc251 std::vector<int64_t> tensor_ids(tensors.size()); in ShouldRecord() local
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddumping_callback.py333 tensor_ids): argument
/aosp_15_r20/external/pytorch/torch/csrc/lazy/python/
H A Dinit.cpp233 std::vector<int64_t> tensor_ids; in initLazyBindings() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Doptional_debug_tools.cc244 void PrintTotalBytesOfTensors(const Subgraph& subgraph, const T& tensor_ids, in PrintTotalBytesOfTensors()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dpywrap_tfe_src.cc1977 std::vector<int64_t> tensor_ids; local
1998 bool TensorShapesAndDtypes(PyObject* tensors, std::vector<int64_t>* tensor_ids,
2040 std::vector<int64_t> tensor_ids; local
2075 std::vector<int64_t> tensor_ids; local
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dlazy_graph_executor.cpp592 std::unordered_set<int64_t> tensor_ids; in CollectSyncTensors() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Ddeadness_analysis.cc1550 std::vector<TensorId> tensor_ids; in Print() local