Searched defs:tensor_indices (Results 1 – 14 of 14) sorted by relevance
55 const int* tensor_indices, in ValidateTensorIndexingSafe()77 int max_size, const int* tensor_indices) { in ValidateTensorIndexing()
55 std::vector<std::vector<int>> tensor_indices; member
137 std::vector<std::vector<int>> tensor_indices; in BuildPerDeviceSave() local
72 const flatbuffers::Vector<int32_t>* tensor_indices, in parseTensorList()
621 const Tensor* tensor_indices; in Compute() local1062 const Tensor* tensor_indices; in Compute() local
33 std::ostream& operator<<(std::ostream& stream, const std::vector<TensorIndex>& tensor_indices) { in operator <<()
46 const TfLiteIntArray* tensor_indices) { in GetTensorNames()
276 const std::vector<int>& tensor_indices, in GetLoggableTensorIndices()
2098 const std::vector<int64_t>& tensor_indices, in compute_bucket_assignment_by_size()
3056 const std::optional<std::shared_ptr<::c10d::Logger>>& logger) { in c10d_init()
4951 std::vector<Value*> tensor_indices; in emitIntAndSliceIndexing() local5153 std::vector<Value*> tensor_indices; in emitMultidimSlicing() local
2913 const TfLiteIntArray* tensor_indices, in IsAllAllowedTensors()
2168 const TfLiteIntArray* tensor_indices, in AppendDynamicDimensions()
3015 tensor_indices, argument