Home
last modified time | relevance | path

Searched refs:allocate_temp (Results 1 – 25 of 83) sorted by relevance

1234

/external/tensorflow/tensorflow/core/kernels/
Ddynamic_partition_op_gpu.cu.cc204 c, c->allocate_temp(DT_INT32, TensorShape({M}), indices_in), done); in AllocateTempSpace()
206 c, c->allocate_temp(DT_INT32, TensorShape({N}), partitions_out), done); in AllocateTempSpace()
208 c, c->allocate_temp(DT_INT32, TensorShape({N}), indices_out), done); in AllocateTempSpace()
249 c->allocate_temp(DT_INT32, TensorShape({num_partitions_}), in ComputeAsync()
264 c->allocate_temp(DT_INT32, TensorShape({num_partitions_}), in ComputeAsync()
286 c->allocate_temp(partition_count.dtype(), partition_count.shape(), in ComputeAsync()
347 c->allocate_temp(DT_INT8, in RadixSort()
384 c->allocate_temp(DT_INT32, TensorShape({num_partitions_}), in CountAndSortParts()
405 c, c->allocate_temp(DT_INT32, TensorShape({1}), &num_runs), done); in CountAndSortParts()
417 c->allocate_temp(DT_INT8, in CountAndSortParts()
Dcudnn_pooling_gpu.cc49 OP_REQUIRES_OK(context, context->allocate_temp( in Compute()
63 context->allocate_temp( in Compute()
138 OP_REQUIRES_OK(context, context->allocate_temp(DataTypeToEnum<T>::value, in Compute()
149 OP_REQUIRES_OK(context, context->allocate_temp(DataTypeToEnum<T>::value, in Compute()
158 context->allocate_temp(DataTypeToEnum<T>::value, in Compute()
167 context->allocate_temp(DataTypeToEnum<T>::value, in Compute()
Dgenerate_vocab_remapping_op.cc111 context, context->allocate_temp(DT_INT64, TensorShape({num_new_vocab_}), in Compute()
119 context, context->allocate_temp( in Compute()
126 context, context->allocate_temp(DT_INT64, TensorShape({num_new_vocab_}), in Compute()
136 context->allocate_temp( in Compute()
Dsubstr_op.cc151 OP_REQUIRES_OK(context, context->allocate_temp( in Compute()
161 context->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
171 context->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
214 OP_REQUIRES_OK(context, context->allocate_temp( in Compute()
224 context->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
234 context->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
Dpooling_ops_common.cc181 OP_REQUIRES_OK(context, context->allocate_temp( in Compute()
194 OP_REQUIRES_OK(context, context->allocate_temp( in Compute()
316 OP_REQUIRES_OK(context, context->allocate_temp(DataTypeToEnum<T>::value, in Compute()
327 OP_REQUIRES_OK(context, context->allocate_temp(DataTypeToEnum<T>::value, in Compute()
336 context->allocate_temp(DataTypeToEnum<T>::value, in Compute()
345 context->allocate_temp(DataTypeToEnum<T>::value, in Compute()
Dquantize_and_dequantize_op.cc86 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in Compute()
88 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in Compute()
147 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in Compute()
149 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in Compute()
Dgpu_device_array.h60 context_->allocate_temp(DT_INT8, TensorShape{total_bytes_}, in Init()
81 TF_RETURN_IF_ERROR(context_->allocate_temp( in Finalize()
Dhistogram_op_gpu.cu.cc49 TF_RETURN_IF_ERROR(context->allocate_temp( in Compute()
89 TF_RETURN_IF_ERROR(context->allocate_temp( in Compute()
Dmultinomial_op.cc99 ctx->allocate_temp(DT_DOUBLE, TensorShape({num_classes}), in operator ()()
192 ctx->allocate_temp( in DoCompute()
197 ctx->allocate_temp( in DoCompute()
202 ctx->allocate_temp(DT_FLOAT, TensorShape({batch_size, num_samples}), in DoCompute()
Ddeep_conv2d.cc428 ctx, ctx->allocate_temp( in operator ()()
445 ctx->allocate_temp( in operator ()()
458 ctx->allocate_temp( in operator ()()
542 ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in operator ()()
967 ctx, ctx->allocate_temp( in operator ()()
985 OP_REQUIRES_OK(ctx, ctx->allocate_temp( in operator ()()
996 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in operator ()()
1065 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in operator ()()
1081 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in operator ()()
1089 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in operator ()()
[all …]
Dbincount_op.cc45 TF_RETURN_IF_ERROR(context->allocate_temp( in Compute()
59 TF_RETURN_IF_ERROR(context->allocate_temp(DataTypeToEnum<T>::value, in Compute()
Dunravel_index_op.cc61 ctx->allocate_temp(DataTypeToEnum<Tidx>::value, in Compute()
72 ctx->allocate_temp(DataTypeToEnum<Tidx>::value, in Compute()
Dfused_batch_norm_op.cc280 OP_REQUIRES_OK(context, context->allocate_temp( in operator ()()
291 OP_REQUIRES_OK(context, context->allocate_temp( in operator ()()
335 context, context->allocate_temp(DataTypeToEnum<U>::value, in operator ()()
420 OP_REQUIRES_OK(context, context->allocate_temp( in operator ()()
432 OP_REQUIRES_OK(context, context->allocate_temp( in operator ()()
444 OP_REQUIRES_OK(context, context->allocate_temp( in operator ()()
691 context->allocate_temp(DataTypeToEnum<U>::value, in Compute()
694 context->allocate_temp(DataTypeToEnum<U>::value, in Compute()
Dsparse_fill_empty_rows_op.cc113 context->allocate_temp(DT_INT64, TensorShape({dense_rows}), in Compute()
157 context->allocate_temp(DT_INT64, TensorShape({dense_rows}), in Compute()
235 OP_REQUIRES_OK(context, context->allocate_temp( in Compute()
Dscale_and_translate_op.cc67 TF_RETURN_IF_ERROR(context->allocate_temp( in ComputeSpansCore()
71 TF_RETURN_IF_ERROR(context->allocate_temp( in ComputeSpansCore()
159 TF_RETURN_IF_ERROR(context->allocate_temp( in ComputeGradSpansCore()
163 TF_RETURN_IF_ERROR(context->allocate_temp( in ComputeGradSpansCore()
349 context, context->allocate_temp(DT_FLOAT, in Compute()
453 OP_REQUIRES_OK(context, context->allocate_temp( in Compute()
Dwhere_op_gpu.cu.h165 TF_RETURN_IF_ERROR(ctx->allocate_temp(
274 TF_RETURN_IF_ERROR(ctx->allocate_temp(DataTypeToEnum<TIndex>::v(),
301 TF_RETURN_IF_ERROR(ctx->allocate_temp(
Dcheck_numerics_op.cc138 OP_REQUIRES_OK(context, context->allocate_temp( in ComputeAsync()
164 context->allocate_temp(DT_INT32, TensorShape({abnormal_detected_size}), in ComputeAsync()
Dtopk_op_gpu.h444 TF_RETURN_IF_ERROR(ctx->allocate_temp(
465 TF_RETURN_IF_ERROR(ctx->allocate_temp(
467 TF_RETURN_IF_ERROR(ctx->allocate_temp(DataTypeToEnum<T>::value,
496 TF_RETURN_IF_ERROR(ctx->allocate_temp(
Dsoftmax_op_gpu.cu.cc152 context->allocate_temp(DataTypeToEnum<T>::value, in Compute()
157 context->allocate_temp(DataTypeToEnum<acc_type>::value, in Compute()
Dgather_nd_op.cc144 c->allocate_temp(DataTypeToEnum<T>::value, result_shape, out)); in DoGatherNd()
162 TF_RETURN_IF_ERROR(c->allocate_temp(DT_INT32, TensorShape(), &scratch)); in DoGatherNd()
Dconv_ops_3d.cc314 ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, transformed_shape, in launch()
333 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in launch()
384 ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in launch()
397 ctx, ctx->allocate_temp( in launch()
/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dgru_ops.cc131 OP_REQUIRES_OK(ctx, ctx->allocate_temp( in Compute()
137 OP_REQUIRES_OK(ctx, ctx->allocate_temp( in Compute()
144 ctx->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
330 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
335 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
340 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
345 OP_REQUIRES_OK(ctx, ctx->allocate_temp( in Compute()
351 OP_REQUIRES_OK(ctx, ctx->allocate_temp( in Compute()
Dlstm_ops.cc343 OP_REQUIRES_OK(ctx, ctx->allocate_temp( in Compute()
350 ctx->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
611 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
616 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
621 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
626 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
631 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
799 TF_CHECK_OK(ctx_->allocate_temp(t.dtype(), t.shape(), &aligned)); in AlignTensor()
946 OP_REQUIRES_OK(ctx, ctx->allocate_temp( in Compute()
953 ctx->allocate_temp(DataTypeToEnum<T>::v(), in Compute()
[all …]
/external/tensorflow/tensorflow/core/util/
Dtensor_ops_util.h39 TF_RETURN_IF_ERROR(ctx->allocate_temp(x.dtype(), x.shape(), out, attr)); in ZerosLikeTensor()
100 TF_RETURN_IF_ERROR(ctx->allocate_temp(a.dtype(), a.shape(), out, attr)); in BinaryAddTensors()
/external/tensorflow/tensorflow/contrib/image/kernels/
Dsegmentation_ops.cc48 OP_REQUIRES_OK(ctx, ctx->allocate_temp(tensorflow::DT_INT64, in Compute()
50 OP_REQUIRES_OK(ctx, ctx->allocate_temp(tensorflow::DT_INT64, in Compute()

1234