Home
last modified time | relevance | path

Searched refs:input_list (Results 1 – 25 of 71) sorted by relevance

123

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dtensor_utils.cc26 const OpInputList& input_list) { in OpInputListToTensorVec() argument
28 tensor_vec.reserve(input_list.size()); in OpInputListToTensorVec()
29 for (const Tensor& tensor : input_list) { in OpInputListToTensorVec()
42 context->input_list(kDenseFloatFeaturesName, features_list)); in ReadDenseFloatFeatures()
57 TF_RETURN_IF_ERROR(context->input_list(kSparseFloatFeatureIndicesName, in ReadSparseFloatFeatures()
60 context->input_list(kSparseFloatFeatureValuesName, feature_values_list)); in ReadSparseFloatFeatures()
62 context->input_list(kSparseFloatFeatureShapesName, feature_shapes_list)); in ReadSparseFloatFeatures()
77 context->input_list(kSparseIntFeatureIndicesName, features_indices_list)); in ReadSparseIntFeatures()
79 context->input_list(kSparseIntFeatureValuesName, feature_values_list)); in ReadSparseIntFeatures()
81 context->input_list(kSparseIntFeatureShapesName, feature_shapes_list)); in ReadSparseIntFeatures()
Dtensor_utils.h31 const OpInputList& input_list);
/external/tensorflow/tensorflow/core/kernels/
Dstring_join_op.cc38 OpInputList input_list; in Compute() local
39 OP_REQUIRES_OK(context, context->input_list("inputs", &input_list)); in Compute()
44 for (const auto& input : input_list) { in Compute()
65 std::vector<StringPiece> strings(input_list.size()); in Compute()
67 for (int j = 0; j < input_list.size(); ++j) { in Compute()
Dlist_kernels.cc196 const TensorList& input_list, in ForwardInputOrCreateNewList() argument
213 output_tensor->scalar<Variant>()() = input_list; in ForwardInputOrCreateNewList()
420 const TensorList* input_list = nullptr; in Compute() local
421 OP_REQUIRES_OK(c, GetInputList(c, 0, &input_list)); in Compute()
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()
442 if (size > input_list->tensors.size()) { in Compute()
444 input_list->tensors.begin(), in Compute()
445 input_list->tensors.end()); in Compute()
[all …]
Dsdca_internal.cc98 context->input_list("sparse_indices", &sparse_indices_inputs)); in Initialize()
101 context->input_list("sparse_weights", &sparse_weights_inputs)); in Initialize()
104 context->input_list("dense_weights", &dense_weights_inputs)); in Initialize()
328 TF_RETURN_IF_ERROR(context->input_list("sparse_example_indices", in Initialize()
331 TF_RETURN_IF_ERROR(context->input_list("sparse_feature_indices", in Initialize()
335 TF_RETURN_IF_ERROR(context->input_list("sparse_feature_values", in Initialize()
357 context->input_list("dense_features", &dense_features_inputs)); in Initialize()
Dsparse_concat_op.cc43 OP_REQUIRES_OK(context, context->input_list("indices", &inds)); in Compute()
53 OP_REQUIRES_OK(context, context->input_list("values", &vals)); in Compute()
65 OP_REQUIRES_OK(context, context->input_list("shapes", &shapes)); in Compute()
Dexample_parsing_ops.cc53 OP_REQUIRES_OK(ctx, ctx->input_list("dense_keys", &dense_keys)); in Compute()
54 OP_REQUIRES_OK(ctx, ctx->input_list("sparse_keys", &sparse_keys)); in Compute()
55 OP_REQUIRES_OK(ctx, ctx->input_list("dense_defaults", &dense_defaults)); in Compute()
183 OP_REQUIRES_OK(ctx, ctx->input_list("dense_defaults", &dense_defaults)); in Compute()
280 OP_REQUIRES_OK(ctx, ctx->input_list("context_dense_defaults", in Compute()
448 ctx->input_list("context_dense_keys", &context_dense_keys)); in Compute()
449 OP_REQUIRES_OK(ctx, ctx->input_list("feature_list_dense_keys", in Compute()
452 ctx, ctx->input_list("context_sparse_keys", &context_sparse_keys)); in Compute()
453 OP_REQUIRES_OK(ctx, ctx->input_list("feature_list_sparse_keys", in Compute()
455 OP_REQUIRES_OK(ctx, ctx->input_list("context_dense_defaults", in Compute()
Dquantized_concat_op.cc170 OP_REQUIRES_OK(context, context->input_list("values", &values)); in Compute()
173 OP_REQUIRES_OK(context, context->input_list("input_mins", &input_mins)); in Compute()
179 OP_REQUIRES_OK(context, context->input_list("input_maxes", &input_maxes)); in Compute()
Dsparse_cross_op.cc302 OP_REQUIRES_OK(context, context->input_list("indices", &indices_list_in)); in Compute()
304 OP_REQUIRES_OK(context, context->input_list("values", &values_list_in)); in Compute()
306 OP_REQUIRES_OK(context, context->input_list("shapes", &shapes_list_in)); in Compute()
309 context->input_list("dense_inputs", &dense_list_in)); in Compute()
/external/vixl/tools/test_generator/
Dparser.py238 input_list = generator.InputList([
243 return operand_list, input_list
320 def ParseTestFile(test_name, test_isa, mnemonics, operand_list, input_list, argument
360 mnemonics, operand_list, input_list, test_cases)
385 operand_list, input_list = ParseDescription(
390 input_list, json_test_file)
/external/tensorflow/tensorflow/core/kernels/data/
Dgraph_rewrite_dataset.cc37 std::vector<std::pair<string, Tensor>> input_list; in Optimize() local
39 params.input_list = &input_list; in Optimize()
72 graph_runner.Run(&graph, lib_, input_list, {output_node}, &outputs)); in Optimize()
89 params.input_list = ctx->input_list(); in AsGraphDefInternal()
Dtensor_dataset_op.cc37 OP_REQUIRES_OK(ctx, ctx->input_list("components", &inputs)); in MakeDataset()
82 DCHECK_NE(ctx->input_list(), nullptr); in AsGraphDefInternal()
83 ctx->input_list()->emplace_back(node->name(), t); in AsGraphDefInternal()
Dtensor_slice_dataset_op.cc39 OP_REQUIRES_OK(ctx, ctx->input_list("components", &inputs)); in MakeDataset()
107 DCHECK_NE(ctx->input_list(), nullptr); in AsGraphDefInternal()
108 ctx->input_list()->emplace_back(node->name(), t); in AsGraphDefInternal()
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_utils.py521 def _get_var_read_dtype(input_list, should_cast): argument
523 if should_cast and input_list and input_list[0].dtype.is_floating:
524 return input_list[0].dtype.base_dtype
529 def autocast_context_manager(input_list, should_cast): argument
544 var_read_dtype = _get_var_read_dtype(input_list, should_cast)
Dbase_layer.py531 input_list = nest.flatten(inputs)
533 if any(isinstance(x, (np.ndarray, float, int)) for x in input_list):
542 input_list = nest.flatten(inputs)
548 build_graph = tf_utils.are_all_symbolic_tensors(input_list)
602 input_list,
653 input_list, self._mixed_precision_policy.should_cast_variables):
1698 input_list = nest.flatten(inputs)
1699 if input_list and self._dtype is None:
1701 self._dtype = input_list[0].dtype.base_dtype.name
1705 if all(hasattr(x, 'shape') for x in input_list):
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc71 OP_REQUIRES_OK(context, context->input_list("node_ids", &node_ids_list)); in Compute()
72 OP_REQUIRES_OK(context, context->input_list("gains", &gains_list)); in Compute()
74 context->input_list("thresholds", &thresholds_list)); in Compute()
75 OP_REQUIRES_OK(context, context->input_list("left_node_contribs", in Compute()
77 OP_REQUIRES_OK(context, context->input_list("right_node_contribs", in Compute()
Dquantile_ops.cc149 context, context->input_list(kFloatFeaturesName, &float_features_list)); in Compute()
236 context->input_list(kSummariesName, &summaries_list)); in Compute()
293 OP_REQUIRES_OK(context, context->input_list(kBucketBoundariesName, in Compute()
453 context, context->input_list(kFloatFeaturesName, &float_features_list)); in Compute()
455 OP_REQUIRES_OK(context, context->input_list(kBucketBoundariesName, in Compute()
Dprediction_ops.cc66 OP_REQUIRES_OK(context, context->input_list("bucketized_features", in Compute()
205 OP_REQUIRES_OK(context, context->input_list("bucketized_features", in Compute()
309 OP_REQUIRES_OK(context, context->input_list("bucketized_features", in Compute()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dchoose_fastest_dataset_op.cc39 OpInputList input_list; in MakeDataset() local
40 OP_REQUIRES_OK(ctx, ctx->input_list("input_datasets", &input_list)); in MakeDataset()
42 ctx, input_list.size() > 1, in MakeDataset()
47 inputs.reserve(input_list.size()); in MakeDataset()
48 for (const auto& tensor : input_list) { in MakeDataset()
/external/tensorflow/tensorflow/python/ops/
Dlist_ops.py279 input_list, _ = op.inputs
280 input_list_size = gen_list_ops.tensor_list_length(input_list)
287 input_list, indices, _ = op.inputs
289 input_list, shape_type=dtypes.int32)
290 num_elements = gen_list_ops.tensor_list_length(input_list)
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dstats_accumulator_ops.cc423 OP_REQUIRES_OK(context, context->input_list("stats_accumulator_handles", in Compute()
427 context->input_list("partition_ids", &partition_ids_list)); in Compute()
431 context->input_list("feature_ids", &feature_ids_list)); in Compute()
433 OP_REQUIRES_OK(context, context->input_list("gradients", &gradients_list)); in Compute()
435 OP_REQUIRES_OK(context, context->input_list("hessians", &hessians_list)); in Compute()
488 OP_REQUIRES_OK(context, context->input_list("stats_accumulator_handles", in Compute()
492 context->input_list("partition_ids", &partition_ids_list)); in Compute()
496 context->input_list("feature_ids", &feature_ids_list)); in Compute()
498 OP_REQUIRES_OK(context, context->input_list("gradients", &gradients_list)); in Compute()
500 OP_REQUIRES_OK(context, context->input_list("hessians", &hessians_list)); in Compute()
Dquantile_ops.cc281 OP_REQUIRES_OK(context, context->input_list(kResourceHandlesName, in Compute()
284 OP_REQUIRES_OK(context, context->input_list(kSummariesName, &summary_list)); in Compute()
693 OP_REQUIRES_OK(context, context->input_list(kResourceHandlesName, in Compute()
881 OP_REQUIRES_OK(context, context->input_list(kDenseValuesName, in Compute()
885 context->input_list(kDenseBucketsName, &dense_buckets_list)); in Compute()
899 context->input_list(kSparseValuesName, in Compute()
903 OP_REQUIRES_OK(context, context->input_list(kSparseIndicesName, in Compute()
908 context, context->input_list(kSparseBucketsName, &sparse_buckets_list)); in Compute()
/external/freetype/src/tools/docmaker/
Dutils.py41 def sort_order_list( input_list, order_list ): argument
43 for id in input_list:
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_ops.cc314 Status GetInitializedList(const xla::XlaOp& input_list, in GetInitializedList() argument
319 IsTensorListInitialized(input_list, &is_already_initialized)); in GetInitializedList()
321 TF_RETURN_IF_ERROR(GetTensorListBufferShape(input_list, &input_list_shape)); in GetInitializedList()
327 TF_RETURN_IF_ERROR(GetTensorListBuffer(input_list, output_list_buffer)); in GetInitializedList()
337 InitializeTensorList(input_list, output_list_shape, &output_list)); in GetInitializedList()
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_graphs.py206 for input_list in self._input_lists:
207 if node_name not in input_list:
209 for inp in input_list[node_name]:

123