| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_op_helpers.h | 49 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in EnsureSparseVariableAccess()
61 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in EnsureSparseVariableAccess()
208 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in PrepareToUpdateVariable()
220 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in PrepareToUpdateVariable()
|
| D | lookup_table_op.h | 46 OP_REQUIRES_OK(ctx, ctx->allocate_persistent(tensorflow::DT_RESOURCE, in LookupTableOp()
50 OP_REQUIRES_OK(ctx, ctx->allocate_persistent(tensorflow::DT_STRING, in LookupTableOp()
|
| D | queue_ops.cc | 81 context->allocate_persistent(DT_STRING, TensorShape({2}), in FakeQueueOp()
|
| D | assign_op.h | 116 context, context->allocate_persistent(old_lhs.dtype(), rhs.shape(), in Compute()
|
| 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 | count_up_to_op.cc | 88 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 | 525 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in LockedWriteOrAggregate()
612 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 | fifo_queue.cc | 95 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 | priority_queue.cc | 123 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in GetElementComponentFromBatch()
|
| D | resource_variable_ops.cc | 400 context->allocate_persistent(value.dtype(), value.shape(), in Compute()
478 context->allocate_persistent(DT_VARIANT, value.shape(), in Compute()
|
| D | random_shuffle_queue_op.cc | 172 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in GetElementComponentFromBatch()
|
| D | lookup_table_op.cc | 727 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in AllocateBuckets()
741 TF_RETURN_IF_ERROR(ctx->allocate_persistent( in AllocateBuckets()
|
| D | tensor_array_ops.cc | 1127 ctx, ctx->allocate_persistent(tensor_array->ElemType(), element_shape, in Compute()
1319 OP_REQUIRES_OK(ctx, ctx->allocate_persistent( in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | zeros_op.h | 72 c->allocate_persistent(DT_INT32, TensorShape({batch_size * (rows + 1)}), in operator()
|
| D | softmax_op.cc | 170 OP_REQUIRES_OK(ctx, 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 | 299 Status allocate_persistent(DataType type, const TensorShape& shape,
1032 Status allocate_persistent(DataType type, const TensorShape& shape,
1035 Status allocate_persistent(DataType type, const TensorShape& shape, in allocate_persistent() function
1038 return allocate_persistent(type, shape, out_persistent, out_tensor, in allocate_persistent()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 450 OP_REQUIRES_OK(context, context->allocate_persistent( in CacheWeight()
464 context, context->allocate_persistent(DataTypeToEnum<Tweight>::value, in CacheWeight()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 559 TF_RETURN_IF_ERROR(context->allocate_persistent( in AllocateOutputTensors()
|