HomeSort by relevance Sort by last modified time
    Searched refs:IncrementKernel (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/
inc_op_kernel.h 26 void IncrementKernel(const float* d_input, float inc, float* d_output,
inc_op_kernel.cu.cc 37 void IncrementKernel(const float* d_input, float inc, float* d_output,
71 IncrementKernel(input_tensor.flat<float>().data(), inc_,
inc_op_plugin.cc 78 IncrementKernel(input, inc_, output, count, stream);

Completed in 281 milliseconds