OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
full:binding_index
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/contrib/tensorrt/kernels/
trt_engine_op.cc
56
size_t
binding_index
;
local
61
binding_index
= trt_engine_ptr_->getBindingIndex(input_nodes_[i].c_str());
71
switch (trt_engine_ptr_->getBindingDataType(
binding_index
)) {
73
buffers[
binding_index
] = (void*)(input_tensor.flat<float>().data());
90
binding_index
= trt_engine_ptr_->getBindingIndex(output_nodes_[i].c_str());
94
if (
binding_index
!= -1) {
95
auto dims = trt_engine_ptr_->getBindingDimensions(
binding_index
);
109
switch (trt_engine_ptr_->getBindingDataType(
binding_index
)) {
111
buffers[
binding_index
] =
/external/tensorflow/tensorflow/contrib/tensorrt/shape_fn/
trt_shfn.cc
66
int
binding_index
= trt_engine->getBindingIndex(output_nodes[i].c_str());
local
70
if (
binding_index
!= -1) {
71
auto dims = trt_engine->getBindingDimensions(
binding_index
);
/external/vulkan-validation-layers/layers/
object_tracker_utils.cpp
[
all
...]
Completed in 89 milliseconds