Home
last modified time | relevance | path

Searched refs:var_tensor (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstateful_random_ops.cc89 Tensor* var_tensor = var->tensor(); in UpdateVariableAndFill() local
90 TF_RETURN_IF_ERROR(CheckState(*var_tensor)); in UpdateVariableAndFill()
91 auto var_tensor_flat = var_tensor->flat<StateElementType>(); in UpdateVariableAndFill()
101 TF_RETURN_IF_ERROR(CheckPhiloxState(*var_tensor, alg_tag_skip)); in UpdateVariableAndFill()
103 ctx, var_tensor, var->copy_on_read_mode.load())); in UpdateVariableAndFill()
109 arg.state_tensor = var_tensor; in UpdateVariableAndFill()
254 Tensor* var_tensor = var->tensor(); in Compute() local
255 OP_REQUIRES_OK(ctx, CheckState(*var_tensor)); in Compute()
257 OP_REQUIRES_OK(ctx, CheckPhiloxState(*var_tensor)); in Compute()
259 ctx, var_tensor, var->copy_on_read_mode.load())); in Compute()
[all …]
Drandom_binomial_op.cc397 Tensor* var_tensor = var->tensor(); in Compute() local
399 ctx, var_tensor->dtype() == STATE_ELEMENT_DTYPE, in Compute()
402 DataTypeString(var_tensor->dtype()))); in Compute()
403 OP_REQUIRES(ctx, var_tensor->dims() == 1, in Compute()
406 var_tensor->dims())); in Compute()
407 auto var_tensor_flat = var_tensor->flat<StateElementType>(); in Compute()
420 ctx, var_tensor, var->copy_on_read_mode.load())); in Compute()
Dresource_variable_ops.cc548 Tensor* var_tensor = variable->tensor(); in Compute() local
549 OP_REQUIRES(context, var_tensor->shape().IsSameSize(value.shape()), in Compute()
551 var_tensor->shape().DebugString(), in Compute()
557 context, var_tensor, variable->copy_on_read_mode.load())); in Compute()
559 update_functor(context->eigen_device<Device>(), var_tensor->flat<T>(), in Compute()
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dtake_while_ops.py44 var_tensor = gen_experimental_dataset_ops.take_while_dataset(
49 super(_TakeWhileDataset, self).__init__(input_dataset, var_tensor)