Home
last modified time | relevance | path

Searched defs:tensors_data (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dlazy_graph_executor.cpp539 std::vector<BackendDataPtr> tensors_data, in Async()
677 std::vector<BackendDataPtr> tensors_data; in SetTensorData() local
883 std::vector<BackendDataPtr> tensors_data, in ScheduleSyncTensorsGraph()
949 auto tensors_data = in ScheduleSyncTensorsGraph() local
966 std::vector<BackendDataPtr> tensors_data = GatherTensorsData( in GetTensorsFused() local
986 c10::ArrayRef<BackendDataPtr> tensors_data, in FetchTensors()
1018 c10::ArrayRef<BackendDataPtr> tensors_data) { in GatherTensorsData()
H A Dlazy_graph_executor.h270 std::map<int64_t, std::weak_ptr<LazyTensor::Data>> tensors_data; member
330 std::vector<BackendDataPtr> tensors_data; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/
H A Dnnapi_plugin_test.cc235 const tflite::TensorData tensors_data = {tflite::TensorType_FLOAT32, in Build() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/
H A Dxrt_state_ops.h314 std::vector<const char*> tensors_data; in Compute() local