| /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()
150 OP_REQUIRES_OK(context, context->allocate_output(0, TensorShape(), in Compute()
155 context->allocate_output(1, TensorShape(), &output_proto_t)); in Compute()
|
| D | prediction_ops.cc | 88 context->allocate_output("partial_logits", in Compute()
94 OP_REQUIRES_OK(context, context->allocate_output("tree_ids", {batch_size}, in Compute()
99 OP_REQUIRES_OK(context, context->allocate_output("node_ids", {batch_size}, in Compute()
216 OP_REQUIRES_OK(context, context->allocate_output( in Compute()
325 context, context->allocate_output("examples_debug_outputs_serialized", in Compute()
|
| /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 | 79 OP_REQUIRES_OK(context, context->allocate_output("empty_row_indicator", in Compute()
85 context, context->allocate_output("reverse_index_map", TensorShape({N}), in Compute()
99 OP_REQUIRES_OK(context, context->allocate_output("output_indices", in Compute()
104 context->allocate_output("output_values", TensorShape({0}), in Compute()
142 OP_REQUIRES_OK(context, context->allocate_output("output_indices", in Compute()
150 context, context->allocate_output( in Compute()
224 OP_REQUIRES_OK(context, context->allocate_output( in Compute()
229 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()
134 context->allocate_output( in ComputeWithReader()
139 context->allocate_output( in ComputeWithReader()
163 OP_REQUIRES_OK(context, context->allocate_output("records_produced", in ComputeWithReader()
181 OP_REQUIRES_OK(context, context->allocate_output("units_completed", in ComputeWithReader()
201 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 | fake_quant_ops.cc | 197 context->allocate_output(0, input.shape(), &output)); in Compute()
239 context->allocate_output(0, input.shape(), &grad_wrt_input)); in Compute()
244 context->allocate_output(1, scalar_shape, &grad_wrt_min)); in Compute()
248 context->allocate_output(2, scalar_shape, &grad_wrt_max)); in Compute()
333 context->allocate_output(0, input.shape(), &output)); in Compute()
383 context->allocate_output(0, input.shape(), &grad_wrt_input)); in Compute()
388 context->allocate_output(1, min_max_shape, &grad_wrt_min)); in Compute()
392 context->allocate_output(2, min_max_shape, &grad_wrt_max)); in Compute()
|
| D | requantize.cc | 48 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, input.shape(), &output)); in Compute()
50 OP_REQUIRES_OK(ctx, ctx->allocate_output(1, TensorShape({}), &output_min)); in Compute()
52 OP_REQUIRES_OK(ctx, ctx->allocate_output(2, TensorShape({}), &output_max)); in Compute()
|
| 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 | 93 ctx, ctx->allocate_output(0, output_shape, &response_t), done); in ComputeAsync()
101 ctx, ctx->allocate_output(1, output_shape, &status_code_t), done); in ComputeAsync()
103 ctx, ctx->allocate_output(2, output_shape, &status_message_t), done); in ComputeAsync()
|
| 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_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 | 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 | 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 | 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/contrib/boosted_trees/kernels/ |
| D | model_ops.cc | 78 OP_REQUIRES_OK(context, context->allocate_output(0, TensorShape(), in Compute()
97 OP_REQUIRES_OK(context, context->allocate_output(0, TensorShape(), in Compute()
102 context, context->allocate_output(1, TensorShape(), &output_config_t)); in Compute()
167 context, context->allocate_output("used_handlers_mask", {num_handlers_}, in Compute()
173 context->allocate_output("num_used_handlers", {}, in Compute()
|
| D | stats_accumulator_ops.cc | 137 OP_REQUIRES_OK(context, context->allocate_output("output_partition_ids", in SerializeScalarAccumulatorToOutput()
144 OP_REQUIRES_OK(context, context->allocate_output("output_feature_ids", in SerializeScalarAccumulatorToOutput()
151 context, context->allocate_output( in SerializeScalarAccumulatorToOutput()
157 context, context->allocate_output("output_hessians", in SerializeScalarAccumulatorToOutput()
178 OP_REQUIRES_OK(context, context->allocate_output("output_partition_ids", in SerializeTensorAccumulatorToOutput()
184 OP_REQUIRES_OK(context, context->allocate_output("output_feature_ids", in SerializeTensorAccumulatorToOutput()
194 context->allocate_output("output_gradients", gradient_shape, in SerializeTensorAccumulatorToOutput()
202 OP_REQUIRES_OK(context, context->allocate_output("output_hessians", in SerializeTensorAccumulatorToOutput()
576 context->allocate_output("num_updates", TensorShape({}), in Compute()
617 context->allocate_output("num_updates", TensorShape({}), in Compute()
[all …]
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/ |
| D | stochastic_hard_routing_gradient_op.cc | 152 OP_REQUIRES_OK(context, context->allocate_output(0, output_routing_shape, in Compute()
155 context, context->allocate_output(1, output_data_shape, &output_data)); in Compute()
156 OP_REQUIRES_OK(context, context->allocate_output(2, output_parameters_shape, in Compute()
159 context, context->allocate_output(3, output_bias_shape, &output_bias)); in Compute()
|
| D | k_feature_gradient_op.cc | 128 context->allocate_output(0, out_routes_shape, &out_routes)); in Compute()
130 context->allocate_output(1, out_data_shape, &out_data)); in Compute()
132 context, context->allocate_output(2, out_weights_shape, &out_weights)); in Compute()
|
| /external/tensorflow/tensorflow/contrib/libsvm/kernels/ |
| D | decode_libsvm_op.cc | 43 ctx, ctx->allocate_output(0, input_tensor->shape(), &label_tensor)); in Compute()
94 OP_REQUIRES_OK(ctx, ctx->allocate_output( in Compute()
119 ctx->allocate_output( in Compute()
126 OP_REQUIRES_OK(ctx, ctx->allocate_output( in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_ops.cc | 44 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({}), &result)); in Compute()
57 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({}), &result)); in Compute()
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops.cc | 313 ctx, ctx->allocate_output("cs", TensorShape({batch_size, cell_size}), in Compute()
318 ctx, ctx->allocate_output("f", TensorShape({batch_size, cell_size}), in Compute()
328 ctx, ctx->allocate_output("ci", TensorShape({batch_size, cell_size}), in Compute()
333 ctx, ctx->allocate_output("co", TensorShape({batch_size, cell_size}), in Compute()
338 ctx, ctx->allocate_output("h", TensorShape({batch_size, cell_size}), in Compute()
590 OP_REQUIRES_OK(ctx, ctx->allocate_output( in Compute()
925 OP_REQUIRES_OK(ctx, ctx->allocate_output("i", batch_cell_shape, &i_out)); in Compute()
928 OP_REQUIRES_OK(ctx, ctx->allocate_output("cs", batch_cell_shape, &cs_out)); in Compute()
931 OP_REQUIRES_OK(ctx, ctx->allocate_output("f", batch_cell_shape, &f_out)); in Compute()
934 OP_REQUIRES_OK(ctx, ctx->allocate_output("o", batch_cell_shape, &o_out)); in Compute()
[all …]
|