Searched refs:src_indexes (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/client/ |
| H A D | tf_session_helper.cc | 692 int* src_indexes; in TF_ImportGraphDefResultsMissingUnusedInputMappings_wrapper() local
694 results, &num_missing_unused_input_mappings, &src_names, &src_indexes); in TF_ImportGraphDefResultsMissingUnusedInputMappings_wrapper()
697 input_strs[i] = TensorId(src_names[i], src_indexes[i]).ToString(); in TF_ImportGraphDefResultsMissingUnusedInputMappings_wrapper()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | literal.cc | 403 DimensionVector src_indexes(src_base.size(), 0); in CopySliceFromInternal() local
411 src_indexes.begin(), std::plus<int64_t>()); in CopySliceFromInternal()
416 int64_t src_index = linear_index(src_literal.shape(), src_indexes); in CopySliceFromInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api.h | 844 const char*** src_names, int** src_indexes);
|
| H A D | c_api_test.cc | 966 int* src_indexes; in TEST() local
968 results, &num_unused_input_mappings, &src_names, &src_indexes); in TEST()
971 EXPECT_EQ(0, src_indexes[0]); in TEST()
|
| H A D | c_api.cc | 1723 const char*** src_names, int** src_indexes) { in TF_ImportGraphDefResultsMissingUnusedInputMappings() argument
1726 *src_indexes = results->missing_unused_key_indexes.data(); in TF_ImportGraphDefResultsMissingUnusedInputMappings()
|