Home
last modified time | relevance | path

Searched refs:index_list (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcross_device_utils.py462 def extract_ranges(index_list, range_size_limit=32): argument
476 if not index_list:
478 first = index_list[0]
482 for i in index_list[1:]:
/external/deqp-deps/SPIRV-Tools/source/opt/
Dir_builder.h408 const std::vector<uint32_t>& index_list) { in AddCompositeExtract() argument
412 for (uint32_t index : index_list) { in AddCompositeExtract()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dir_builder.h408 const std::vector<uint32_t>& index_list) { in AddCompositeExtract() argument
412 for (uint32_t index : index_list) { in AddCompositeExtract()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3335 TRT_ShapedWeights index_list = inputs.at(1).weights(); in ConvertReduce() local
3344 if (index_list.count() == 0) { in ConvertReduce()
3350 static_cast<int*>(const_cast<void*>(index_list.GetValues())); in ConvertReduce()
3351 for (int i = 0; i < index_list.count(); i++) { in ConvertReduce()