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

  /external/tensorflow/tensorflow/contrib/framework/kernels/
zero_initializer_op_gpu.cu.cc 29 #define DEFINE_GPU_SPECS(T) template struct TensorSetZero<GPUDevice, T>;
zero_initializer_op.h 25 struct TensorSetZero {
zero_initializer_op.cc 53 functor::TensorSetZero<Device, T>()(ctx->eigen_device<Device>(),
73 void TensorSetZero<GPUDevice, T>::operator()(const GPUDevice& d, \
75 extern template struct TensorSetZero<GPUDevice, T>;
  /external/tensorflow/tensorflow/core/kernels/
tensor_array.cc 58 Status TensorSetZero<Device, T>(OpKernelContext * ctx, Tensor * value) { \
tensor_array.h 73 Status TensorSetZero(OpKernelContext* ctx, Tensor* value) {
75 "tensor_array::TensorSetZero type not supported: ",
81 Status TensorSetZero<Device, T>(OpKernelContext * ctx, Tensor * value);
612 Status s = tensor_array::TensorSetZero<Device, T>(ctx, tensor_t);

Completed in 76 milliseconds