Home | Sort by relevance Sort by last modified time |
/external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ | |
inc_op_plugin.h | 84 std::memcpy(buffer, &inc_, sizeof(float)); 89 float inc_; member in class:tensorflow::tensorrt::IncOpPlugin |
inc_op_kernel.cu.cc | 57 inc_ = inc_list[0]; 71 IncrementKernel(input_tensor.flat<float>().data(), inc_, 77 float inc_; member in class:tensorflow::tensorrt::IncPluginTRT |