Home
last modified time | relevance | path

Searched refs:node_indices (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/compiler/
Dcompiled_node.cc56 std::move(attr->node_indices.begin(), attr->node_indices.end(), in MergeCode()
57 std::back_inserter(merged_attr->node_indices)); in MergeCode()
Dshader_code.h43 node_indices(in_node_indices) {} in ShaderCode()
61 std::vector<NodeId> node_indices; member
Dcompiled_node.h39 std::vector<NodeId> node_indices; member
Dshader_codegen.cc178 attr.code.workgroup, partial_source_code, attr.node_indices); in Build()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/workgroups/
Dcalculator_from_metadata.cc51 for (auto node_id : *workgroup->node_indices()) { in WorkgroupsCalculatorFromMetadata()
58 auto it = workgroups_.find(shader_code.node_indices[0]); in CalculateInternal()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dworkgroups.fbs28 node_indices:[uint32];
Dcompiler.cc124 attr.node_indices.push_back(node->id); in Compile()
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py1181 node_indices = nodes.keys()
1182 node_indices = sorted(node_indices)
1183 return node_indices[-1]