| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_activation_ops.cc | 39 context->allocate_output(0, input.shape(), &output)); in Compute()
52 OP_REQUIRES_OK(context, context->allocate_output(1, {}, &output_min)); in Compute()
55 OP_REQUIRES_OK(context, context->allocate_output(2, {}, &output_max)); in Compute()
72 context->allocate_output(0, input.shape(), &output)); in Compute()
90 OP_REQUIRES_OK(context, context->allocate_output(1, {}, &output_min)); in Compute()
93 OP_REQUIRES_OK(context, context->allocate_output(2, {}, &output_max)); in Compute()
|
| D | sparse_fill_empty_rows_op.cc | 84 OP_REQUIRES_OK(context, context->allocate_output(kEmptyRowIndicatorOutput, in Compute()
89 OP_REQUIRES_OK(context, context->allocate_output(kReverseIndexMapOutput, in Compute()
104 OP_REQUIRES_OK(context, context->allocate_output(kOutputIndicesOutput, in Compute()
108 OP_REQUIRES_OK(context, context->allocate_output(kOutputValuesOutput, in Compute()
147 OP_REQUIRES_OK(context, context->allocate_output(kOutputIndicesOutput, in Compute()
154 OP_REQUIRES_OK(context, context->allocate_output(kOutputValuesOutput, in Compute()
229 OP_REQUIRES_OK(context, context->allocate_output( in Compute()
234 context->allocate_output("d_default_value", TensorShape({}), in Compute()
|
| D | reader_ops.cc | 88 context->allocate_output("key", TensorShape({}), &key)); in ComputeWithReader()
91 context->allocate_output("value", TensorShape({}), &value)); in ComputeWithReader()
137 context->allocate_output( in ComputeWithReader()
142 context->allocate_output( in ComputeWithReader()
166 OP_REQUIRES_OK(context, context->allocate_output("records_produced", in ComputeWithReader()
184 OP_REQUIRES_OK(context, context->allocate_output("units_completed", in ComputeWithReader()
204 context->allocate_output("state", TensorShape({}), &output)); in ComputeWithReader()
|
| D | quantized_pooling_ops.cc | 74 OP_REQUIRES_OK(context, context->allocate_output( in Compute()
95 OP_REQUIRES_OK(context, context->allocate_output(1, {}, &output_min)); in Compute()
98 OP_REQUIRES_OK(context, context->allocate_output(2, {}, &output_max)); in Compute()
119 OP_REQUIRES_OK(context, context->allocate_output(1, {}, &output_min)); in Compute()
122 OP_REQUIRES_OK(context, context->allocate_output(2, {}, &output_max)); in Compute()
|
| D | candidate_sampler_ops.cc | 62 context->allocate_output(0, TensorShape({num_sampled_}), in Compute()
66 OP_REQUIRES_OK(context, context->allocate_output( in Compute()
71 context->allocate_output(2, TensorShape({num_sampled_}), in Compute()
240 context->allocate_output( in Compute()
244 context, context->allocate_output( in Compute()
249 context->allocate_output( in Compute()
|
| D | shape_ops.h | 49 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({rank}), &out)); in Compute()
77 OP_REQUIRES_OK(ctx, ctx->allocate_output(i, {dims}, &out)); in Compute()
105 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({}), &out)); in Compute()
122 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({}), &out)); in Compute()
174 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, {0}, &output)); in Compute()
245 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, {0}, &output)); in Compute()
|
| D | requantize.cc | 47 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, input.shape(), &output)); in Compute()
49 OP_REQUIRES_OK(ctx, ctx->allocate_output(1, TensorShape({}), &output_min)); in Compute()
51 OP_REQUIRES_OK(ctx, ctx->allocate_output(2, TensorShape({}), &output_max)); in Compute()
|
| D | fake_quant_ops.cc | 209 context->allocate_output(0, input.shape(), &output)); in Compute()
251 context->allocate_output(0, input.shape(), &grad_wrt_input)); in Compute()
256 context->allocate_output(1, scalar_shape, &grad_wrt_min)); in Compute()
260 context->allocate_output(2, scalar_shape, &grad_wrt_max)); in Compute()
347 context->allocate_output(0, input.shape(), &output)); in Compute()
397 context->allocate_output(0, input.shape(), &grad_wrt_input)); in Compute()
402 context->allocate_output(1, min_max_shape, &grad_wrt_min)); in Compute()
406 context->allocate_output(2, min_max_shape, &grad_wrt_max)); in Compute()
|
| D | debug_ops.h | 82 OP_REQUIRES_OK(context, context->allocate_output(0, src_tensor.shape(), in Compute()
180 if (!context->allocate_output(0, shape, &output_tensor).ok()) { in ApplyGrpcGating()
269 OP_REQUIRES_OK(context, context->allocate_output(0, shape, &output_tensor)); in Compute()
372 OP_REQUIRES_OK(context, context->allocate_output(0, shape, &output_tensor)); in Compute()
545 context->allocate_output(0, shape, &output_tensor));
558 context->allocate_output(0, shape, &output_tensor));
583 context->allocate_output(0, shape, &output_tensor));
622 context->allocate_output(0, shape, &output_tensor));
648 context->allocate_output(0, shape, &output_tensor));
731 context->allocate_output(0, shape, &output_tensor));
[all …]
|
| D | quantize_down_and_shrink_range.cc | 46 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, input.shape(), &output)); in Compute()
48 OP_REQUIRES_OK(ctx, ctx->allocate_output(1, TensorShape({}), &output_min)); in Compute()
50 OP_REQUIRES_OK(ctx, ctx->allocate_output(2, TensorShape({}), &output_max)); in Compute()
|
| D | rpc_op.cc | 92 ctx, ctx->allocate_output(0, output_shape, &response_t), done); in ComputeAsync()
100 ctx, ctx->allocate_output(1, output_shape, &status_code_t), done); in ComputeAsync()
102 ctx, ctx->allocate_output(2, output_shape, &status_message_t), done); in ComputeAsync()
|
| D | quantized_bias_add_op.cc | 62 context->allocate_output(0, input.shape(), &output)); in Compute()
85 OP_REQUIRES_OK(context, context->allocate_output(1, {}, &output_min)); in Compute()
89 OP_REQUIRES_OK(context, context->allocate_output(2, {}, &output_max)); in Compute()
|
| D | string_split_op.cc | 214 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({output_size, 2}), in Compute()
218 ctx, ctx->allocate_output(1, TensorShape({output_size}), &sp_tokens_t)); in Compute()
220 OP_REQUIRES_OK(ctx, ctx->allocate_output(2, TensorShape({2}), &sp_shape_t)); in Compute()
284 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({output_size, 2}), in Compute()
288 ctx, ctx->allocate_output(1, TensorShape({output_size}), &sp_tokens_t)); in Compute()
290 OP_REQUIRES_OK(ctx, ctx->allocate_output(2, TensorShape({2}), &sp_shape_t)); in Compute()
|
| D | quantized_reshape_op.cc | 37 OP_REQUIRES_OK(ctx, ctx->allocate_output(1, TensorShape({}), &output_min)); in Compute()
41 OP_REQUIRES_OK(ctx, ctx->allocate_output(2, TensorShape({}), &output_max)); in Compute()
|
| D | quantize_op.cc | 121 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, input.shape(), &output)); in Compute()
127 OP_REQUIRES_OK(ctx, ctx->allocate_output(1, {}, &output_min_tensor)); in Compute()
128 OP_REQUIRES_OK(ctx, ctx->allocate_output(2, {}, &output_max_tensor)); in Compute()
140 ctx->allocate_output(1, minmax_shape, &output_min_tensor)); in Compute()
142 ctx->allocate_output(2, minmax_shape, &output_max_tensor)); in Compute()
|
| D | base64_ops.cc | 36 OP_REQUIRES_OK(context, context->allocate_output(0, input_tensor.shape(), in Compute()
61 OP_REQUIRES_OK(context, context->allocate_output(0, input_tensor.shape(), in Compute()
|
| D | requantization_range_op.cc | 52 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({}), &output_min)); in Compute()
54 OP_REQUIRES_OK(ctx, ctx->allocate_output(1, TensorShape({}), &output_max)); in Compute()
|
| D | list_kernels.cc | 122 c->allocate_output(output_index, {}, &output_tensor, attr)); in ForwardInputOrCreateNewList()
145 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape{}, &result, attr)); in Compute()
238 OP_REQUIRES_OK(c, c->allocate_output(0, TensorShape{}, &result)); in Compute()
263 OP_REQUIRES_OK(c, c->allocate_output(0, TensorShape({}), &result)); in Compute()
270 OP_REQUIRES_OK(c, c->allocate_output( in Compute()
312 OP_REQUIRES_OK(c, c->allocate_output(0, TensorShape{}, &result, attr)); in Compute()
362 OP_REQUIRES_OK(c, c->allocate_output(0, TensorShape{}, &result, attr)); in Compute()
493 OP_REQUIRES_OK(c, c->allocate_output(0, tl_a_shape, &output, attr)); in Compute()
|
| D | summary_op.cc | 60 OP_REQUIRES_OK(c, c->allocate_output(0, TensorShape({}), &summary_tensor)); in Compute()
109 OP_REQUIRES_OK(c, c->allocate_output(0, TensorShape({}), &summary_tensor)); in Compute()
166 OP_REQUIRES_OK(c, c->allocate_output(0, TensorShape({}), &summary_tensor)); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | resource_ops.cc | 104 OP_REQUIRES_OK(context, context->allocate_output(0, TensorShape(), in Compute()
106 OP_REQUIRES_OK(context, context->allocate_output(1, TensorShape(), in Compute()
109 context->allocate_output(2, TensorShape(), in Compute()
112 context->allocate_output(3, TensorShape(), in Compute()
114 OP_REQUIRES_OK(context, context->allocate_output( in Compute()
149 OP_REQUIRES_OK(context, context->allocate_output(0, TensorShape(), in Compute()
154 context->allocate_output(1, TensorShape(), &output_proto_t)); in Compute()
|
| D | stats_ops.cc | 352 OP_REQUIRES_OK(context, context->allocate_output("node_ids", {num_nodes}, in Compute()
358 OP_REQUIRES_OK(context, context->allocate_output("gains", {num_nodes}, in Compute()
365 context->allocate_output("feature_dimensions", {num_nodes}, in Compute()
372 OP_REQUIRES_OK(context, context->allocate_output("thresholds", {num_nodes}, in Compute()
378 OP_REQUIRES_OK(context, context->allocate_output( in Compute()
386 OP_REQUIRES_OK(context, context->allocate_output( in Compute()
395 context, context->allocate_output("split_with_default_directions", in Compute()
729 OP_REQUIRES_OK(context, context->allocate_output("node_ids", {num_nodes}, in Compute()
735 OP_REQUIRES_OK(context, context->allocate_output("gains", {num_nodes}, in Compute()
741 OP_REQUIRES_OK(context, context->allocate_output("feature_ids", {num_nodes}, in Compute()
[all …]
|
| D | prediction_ops.cc | 95 context->allocate_output("partial_logits", in Compute()
101 OP_REQUIRES_OK(context, context->allocate_output("tree_ids", {batch_size}, in Compute()
106 OP_REQUIRES_OK(context, context->allocate_output("node_ids", {batch_size}, in Compute()
227 OP_REQUIRES_OK(context, context->allocate_output( in Compute()
332 context, context->allocate_output("examples_debug_outputs_serialized", in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | random_seed_ops.cc | 68 OP_REQUIRES_OK(ctx, ctx->allocate_output(kNumRandomSamples, TensorShape({}), in Serialize()
72 OP_REQUIRES_OK(ctx, ctx->allocate_output(kSeed, TensorShape({}), &seed)); in Serialize()
75 OP_REQUIRES_OK(ctx, ctx->allocate_output(kSeed2, TensorShape({}), &seed2)); in Serialize()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_matrix_components_op.cc | 78 c, c->allocate_output(0, TensorShape({rows + 1}), &row_ptrs_t)); in Compute()
79 OP_REQUIRES_OK(c, c->allocate_output(1, TensorShape({nnz}), &col_inds_t)); in Compute()
80 OP_REQUIRES_OK(c, c->allocate_output(2, TensorShape({nnz}), &values_t)); in Compute()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | gpu_swapping_kernels.cc | 42 ctx, ctx->allocate_output(0, input.shape(), &output, alloc_attrs), in ComputeAsync()
71 OP_REQUIRES_OK_ASYNC(ctx, ctx->allocate_output(0, input.shape(), &output), in ComputeAsync()
|