Home
last modified time | relevance | path

Searched refs:input_labels (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/python/
Dtensor_forest_test.py70 input_labels = [0, 1, 2, 3]
80 graph = graph_builder.training_graph(input_data, input_labels)
86 input_labels = [0, 1, 2, 3]
97 graph = graph_builder.training_graph(input_data, input_labels)
163 input_labels = [0, 1, 2, 3]
173 graph = graph_builder.training_graph(input_data, input_labels)
Dtensor_forest.py406 input_labels, argument
429 if input_labels is not None:
430 labels = data_ops.ParseLabelTensorOrDict(input_labels)
584 input_labels, argument
636 input_labels,
647 input_labels,
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops.py241 for input_labels in input_axis_labels:
242 if (len(input_axis_labels) == 1 and input_labels.count(a) == 2 and
243 input_labels == input_labels[::-1] and '->' not in equation):
245 if input_labels.count(a) > 1:
248 input_labels)
/external/autotest/utils/
Dlabellib_unittest.py70 input_labels = copy.deepcopy(labels)
71 mapping = labellib.LabelsMapping(input_labels)
73 self.assertEqual(input_labels, labels)
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dmodel_ops.cc336 const Tensor& input_labels = context->input(2); in Compute() local
345 const int num_data = input_labels.shape().dim_size(0); in Compute()
347 input_labels.shape().dims() <= 1 in Compute()
349 : static_cast<int>(input_labels.shape().dim_size(1)); in Compute()
353 TensorInputTarget target(input_labels, input_weights, num_targets); in Compute()
Dstats_ops.cc245 const Tensor& input_labels = context->input(6); in Compute() local
294 input_labels.shape().dims() <= 1 in Compute()
296 : static_cast<int>(input_labels.shape().dim_size(1)); in Compute()
304 TensorInputTarget target(input_labels, input_weights, num_targets); in Compute()
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/kernel_tests/
Drouting_function_op_test.py32 self.input_labels = [0., 1., 2., 3.]
Dk_feature_routing_function_op_test.py33 self.input_labels = [0., 1., 2., 3.]