Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_configuration_ops.cc138 Tensor* ctx_output; in Compute() local
139 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({}), &ctx_output)); in Compute()
140 ctx_output->scalar<tstring>()() = std::move(host_config_output); in Compute()
227 Tensor* ctx_output; in Compute() local
228 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({}), &ctx_output)); in Compute()
229 ctx_output->scalar<tstring>()() = in Compute()
357 Tensor* ctx_output; in Compute() local
361 &ctx_output)); in Compute()
364 ctx_output->flat<int32>()(i) = device_id_output[i]; in Compute()
396 Tensor* ctx_output; in Compute() local
[all …]