| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 111 xla::XlaOp output_list; in Compile() local
115 &output_list)); in Compile()
116 ctx->SetTensorListOutput(0, output_list); in Compile()
148 xla::XlaOp output_list; in Compile() local
151 &output_list)); in Compile()
152 ctx->SetTensorListOutput(0, output_list); in Compile()
294 xla::XlaOp output_list; in Compile() local
297 &output_list)); in Compile()
298 ctx->SetTensorListOutput(0, output_list); in Compile()
335 xla::XlaOp output_list; in GetInitializedList() local
[all …]
|
| D | tensor_list_utils.cc | 32 xla::XlaOp* output_list) { in BuildTensorList() argument
34 *output_list = xla::Tuple(buffer.builder(), {buffer, push_index}); in BuildTensorList()
76 xla::XlaOp* output_list) { in InitializeTensorList() argument
97 return BuildTensorList(new_buffer, push_index, output_list); in InitializeTensorList()
|
| D | tensor_list_utils.h | 36 xla::XlaOp* output_list);
63 xla::XlaOp* output_list);
|
| /external/mesa3d/src/gallium/drivers/swr/rasterizer/codegen/ |
| D | gen_backends.py | 72 output_list = []
74 output_list.append(list(range(x)))
77 output_combinations = list(itertools.product(*output_list))
78 output_list = []
91 output_list.append(tempStr)
97 numFiles = (len(output_list) + args.split - 1) // args.split
100 linesPerFile = (len(output_list) + numFiles - 1) // numFiles
101 chunkedList = [output_list[x:x+linesPerFile] for x in range(0, len(output_list), linesPerFile)]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops.cc | 150 OP_REQUIRES_OK(ctx, ctx->output_list("dense_values", &dense_values)); in Compute()
151 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_indices", &sparse_indices)); in Compute()
152 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_values", &sparse_values)); in Compute()
153 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_shapes", &sparse_shapes)); in Compute()
246 OP_REQUIRES_OK(ctx, ctx->output_list("dense_values", &dense_values)); in Compute()
247 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_indices", &sparse_indices)); in Compute()
248 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_values", &sparse_values)); in Compute()
249 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_shapes", &sparse_shapes)); in Compute()
378 OP_REQUIRES_OK(ctx, ctx->output_list("context_sparse_indices", in Compute()
381 ctx, ctx->output_list("context_sparse_values", &context_sparse_values)); in Compute()
[all …]
|
| D | list_kernels.cc | 197 TensorList** output_list) { in ForwardInputOrCreateNewList() argument
215 *output_list = output_tensor->scalar<Variant>()().get<TensorList>(); in ForwardInputOrCreateNewList()
302 TensorList* output_list = nullptr; in Compute() local
303 OP_REQUIRES_OK(c, ForwardInputOrCreateNewList(c, 0, 0, *l, &output_list)); in Compute()
304 output_list->tensors.push_back(input); in Compute()
438 TensorList output_list; in Compute() local
439 output_list.element_shape = input_list->element_shape; in Compute()
440 output_list.element_dtype = input_list->element_dtype; in Compute()
441 output_list.max_num_elements = input_list->max_num_elements; in Compute()
443 output_list.tensors.insert(output_list.tensors.begin(), in Compute()
[all …]
|
| D | list_kernels.h | 78 TensorList** output_list);
289 TensorList* output_list = nullptr; in Compute() local
290 OP_REQUIRES_OK(c, ForwardInputOrCreateNewList(c, 0, 0, *l, &output_list)); in Compute()
291 output_list->tensors.pop_back(); in Compute()
511 TensorList output_list; in Compute() local
513 output_list.element_dtype = input_tensor.dtype(); in Compute()
533 output_list.element_shape = element_shape; in Compute()
539 output_list.tensors.reserve(lengths.shape().dim_size(0)); in Compute()
560 output_list.tensors.emplace_back(aligned); in Compute()
566 output_tensor->scalar<Variant>()() = std::move(output_list); in Compute()
[all …]
|
| D | scoped_allocator_ops_test.cc | 180 OpOutputList output_list; in ExecOp() local
181 Status s = context_->output_list("output", &output_list); in ExecOp()
183 const Tensor& output = *(output_list[0]); in ExecOp()
302 OpOutputList output_list; in ExecOp() local
303 Status s = context_->output_list("output", &output_list); in ExecOp()
305 for (int i = 0; i < output_list.size(); i++) { in ExecOp()
306 const Tensor& output = *(output_list[i]); in ExecOp()
|
| D | remote_fused_graph_execute_op_test.cc | 103 const string& name, const std::vector<Output>& output_list, in BuildRemoteFusedGraphExecuteOp() argument
109 auto node_out_list = ops::AsNodeOutList(scope, InputList(output_list)); in BuildRemoteFusedGraphExecuteOp()
254 std::vector<Output> output_list; in RewriteGraphToFusedGraph() local
256 output_list.emplace_back(op_a); in RewriteGraphToFusedGraph()
259 BuildRemoteFusedGraphExecuteOp(REMOTE_FUSED_EXECUTE_OP_NODE_NAME, output_list, in RewriteGraphToFusedGraph()
|
| D | queue_op.cc | 185 ctx, ctx->output_list("components", &output_components), callback); in ComputeAsync()
240 ctx, ctx->output_list("components", &output_components), callback); in ComputeAsync()
313 ctx, ctx->output_list("components", &output_components), callback); in ComputeAsync()
|
| D | ctc_decoder_ops.cc | 102 s = ctx->output_list("decoded_indices", decoded_indices); in ValidateInputsGenerateOutputs()
104 s = ctx->output_list("decoded_values", decoded_values); in ValidateInputsGenerateOutputs()
106 s = ctx->output_list("decoded_shape", decoded_shape); in ValidateInputsGenerateOutputs()
|
| D | identity_n_op.h | 31 OP_REQUIRES_OK(context, context->output_list("output", &output)); in Compute()
|
| D | dynamic_partition_op.cc | 69 OP_REQUIRES_OK(c, c->output_list("outputs", Tout)); in ValidateAndAllocateOutputs()
|
| D | sdca_internal.cc | 107 TF_RETURN_IF_ERROR(context->output_list("out_delta_sparse_weights", in Initialize()
112 context->output_list("out_delta_dense_weights", &dense_weights_outputs)); in Initialize()
|
| D | ragged_gather_op.cc | 217 context->output_list("output_nested_splits", &splits_out)); in WriteSplits()
|
| D | decode_csv_op.cc | 77 OP_REQUIRES_OK(ctx, ctx->output_list("output", &output)); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 68 context, context->output_list("node_ids_list", &output_node_ids_list)); in Compute()
71 context->output_list("gains_list", &output_gains_list)); in Compute()
73 OP_REQUIRES_OK(context, context->output_list("thresholds_list", in Compute()
77 context->output_list("left_node_contribs_list", in Compute()
81 context->output_list("right_node_contribs_list", in Compute()
|
| D | quantile_ops.cc | 169 context, context->output_list(kSummariesName, &summaries_output_list)); in Compute()
404 OP_REQUIRES_OK(context, context->output_list(kBucketBoundariesName, in Compute()
463 OP_REQUIRES_OK(context, context->output_list(kBucketsName, &buckets_list)); in Compute()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 107 OpOutputList output_list; in QuantizeFeatures() local
108 OP_REQUIRES_OK(context, context->output_list(output_name, &output_list)); in QuantizeFeatures()
118 context, output_list.allocate(feature_index, in QuantizeFeatures()
198 OpOutputList* output_list) { in CopyBoundaries() argument
202 context, output_list->allocate( in CopyBoundaries()
382 context->output_list(kSparseSummariesName, in Compute()
385 OP_REQUIRES_OK(context, context->output_list(kDenseSummariesName, in Compute()
391 OpOutputList* output_list) { in Compute() argument
400 OP_REQUIRES_OK(context, output_list->allocate(index, {}, &output_t)); in Compute()
696 OP_REQUIRES_OK(context, context->output_list(kAreBucketsReadyName, in Compute()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transfer_utils.cc | 115 std::vector<Output> output_list; in BuildFusedGraphDef() local
127 output_list.emplace_back(Output(ret, 0)); in BuildFusedGraphDef()
147 auto node_out_list = ops::AsNodeOutList(scope, InputList(output_list)); in BuildFusedGraphDef()
|
| /external/toolchain-utils/cros_utils/ |
| D | buildbot_utils.py | 74 output_list = ast.literal_eval(msg)
75 output_dict = output_list[0]
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sdca_ops.py | 289 output_list = []
295 output_list.append(internal_convert_to_tensor(
297 return output_list
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_linking_helpers.c | 355 struct exec_list *output_list = &producer->outputs; in compact_components() local
469 remap_slots_and_components(output_list, producer->info.stage, remap, in compact_components()
|
| /external/tensorflow/tensorflow/lite/tools/accuracy/ |
| D | run_tflite_model_op.cc | 92 context->output_list("model_output", &output_tensors)); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 303 TF_RETURN_IF_ERROR(ctx->output_list("output", &opts->output)); in SetupOutputs()
|