| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_op_helpers.h | 48 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in EnsureSparseVariableAccess()
60 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in EnsureSparseVariableAccess()
209 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in PrepareToUpdateVariable()
221 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in PrepareToUpdateVariable()
|
| D | queue_ops.cc | 63 context->allocate_persistent(DT_STRING, TensorShape({2}), in FakeQueueOp()
|
| D | barrier_ops.cc | 344 TF_RETURN_IF_ERROR(ctx->allocate_persistent(DT_INT64, TensorShape({}), in InsertOneLocked()
370 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in InsertOneLocked()
391 TF_RETURN_IF_ERROR(ctx->allocate_persistent(DT_STRING, TensorShape({}), in InsertOneLocked()
|
| D | assign_op.h | 116 context, context->allocate_persistent(old_lhs.dtype(), rhs.shape(), in Compute()
|
| D | count_up_to_op.cc | 89 OP_REQUIRES_OK(context, context->allocate_persistent( in Compute()
|
| D | conditional_accumulator.h | 91 ctx->allocate_persistent(dtype_, grad->shape(), &accum_grad_persistent_, in AllocateAndAssignToAccumGradFunction()
|
| D | dense_update_functor.cc | 91 TF_RETURN_IF_ERROR(context->allocate_persistent( \
|
| D | sparse_conditional_accumulator.h | 176 ctx->allocate_persistent(dtype_, grad_val->shape(), accum_val_persistent_, in AllocateAndAssignToAccumGradFunction()
262 ctx, ctx->allocate_persistent(dtype_, sum_shape, tensor_sum_persistent, in AddToAccumGradFunction()
|
| D | tensor_array.h | 526 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in LockedWriteOrAggregate()
613 ctx->allocate_persistent(dtype_, t.shape, &t.tensor, &tensor_t)); in LockedRead()
|
| D | conditional_accumulator_base_op.h | 50 context->allocate_persistent(DT_STRING, TensorShape({2}), in ConditionalAccumulatorBaseOp()
|
| D | sparse_tensors_map_ops.cc | 58 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in AddSparseTensor()
64 TF_RETURN_IF_ERROR(ctx->allocate_persistent(sp.indices().dtype(), in AddSparseTensor()
|
| D | lookup_table_op.h | 44 OP_REQUIRES_OK(ctx, ctx->allocate_persistent(tensorflow::DT_STRING, in LookupTableOp()
|
| D | fifo_queue.cc | 95 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in GetElementComponentFromBatch()
|
| D | priority_queue.cc | 123 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in GetElementComponentFromBatch()
|
| D | padding_fifo_queue.cc | 63 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in GetElementComponent()
|
| D | resource_variable_ops.cc | 369 context->allocate_persistent(value.dtype(), value.shape(), in Compute()
448 context->allocate_persistent(DT_VARIANT, value.shape(), in Compute()
|
| D | random_shuffle_queue_op.cc | 171 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in GetElementComponentFromBatch()
|
| D | lookup_table_op.cc | 705 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in AllocateBuckets()
719 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in AllocateBuckets()
|
| D | tensor_array_ops.cc | 1134 ctx, ctx->allocate_persistent(tensor_array->ElemType(), element_shape, in Compute()
1327 OP_REQUIRES_OK(ctx, ctx->allocate_persistent( in Compute()
|
| D | functional_ops.cc | 651 OP_REQUIRES_OK(context, context->allocate_persistent( in FakeParamOp()
|
| /external/tensorflow/tensorflow/contrib/framework/kernels/ |
| D | zero_initializer_op.cc | 52 ctx, ctx->allocate_persistent(input.dtype(), input.shape(), in Compute()
108 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_op_kernel.h | 52 context->allocate_persistent(DT_STRING, TensorShape({2}), in ResourceOpKernel()
|
| D | op_kernel.h | 305 Status allocate_persistent(DataType type, const TensorShape& shape,
988 Status allocate_persistent(DataType type, const TensorShape& shape,
991 Status allocate_persistent(DataType type, const TensorShape& shape, in allocate_persistent() function
994 return allocate_persistent(type, shape, out_persistent, out_tensor, in allocate_persistent()
|
| D | op_kernel.cc | 254 Status OpKernelConstruction::allocate_persistent( in allocate_persistent() function in tensorflow::OpKernelConstruction
758 Status OpKernelContext::allocate_persistent(DataType type, in allocate_persistent() function in tensorflow::OpKernelContext
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 670 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in AllocateCalibrationResources()
|