OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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