HomeSort by relevance Sort by last modified time
    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