HomeSort by relevance Sort by last modified time
    Searched defs:Toutput (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
inplace_ops_functor_gpu.cu.cc 47 auto Toutput = output->flat_outer_dims<T>();
48 const int64 nrows = Toutput.dimension(0);
49 const int64 ncols = Toutput.dimension(1);
depthtospace_op.cc 113 auto Toutput = outputs_tensor->tensor<T, kDims>();
127 Toutput);
137 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput);
spacetodepth_op.cc 115 auto Toutput = outputs_tensor->tensor<T, kDims>();
129 Toutput);
139 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput);
inplace_ops.cc 38 auto Toutput = output->flat_outer_dims<T>();
39 auto nrows = Toutput.dimension(0);
41 Toutput.template chip<0>(r).device(d) = Tvalue.template chip<0>(0);

Completed in 764 milliseconds