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

  /external/tensorflow/tensorflow/core/kernels/
depthtospace_op.cc 112 auto Tinput = input.tensor<T, kDims>();
126 functor(context->eigen_device<GPUDevice>(), Tinput, block_size_,
137 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput);
spacetodepth_op.cc 114 auto Tinput = input.tensor<T, kDims>();
128 functor(context->eigen_device<GPUDevice>(), Tinput, block_size_,
139 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput);
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 273 milliseconds