| /external/tensorflow/tensorflow/lite/kernels/ |
| D | topk_v2.cc | 58 TfLiteTensor* output_values = GetOutput(context, node, kOutputValues); in ResizeOutput() local
61 output_values->type = input->type; in ResizeOutput()
75 resize_tensor(output_values, output_values_shape, nullptr)); in ResizeOutput()
139 int32* output_indexes, T* output_values) { in TopK() argument
150 T* output_row = output_values + row * k; in TopK()
167 TfLiteTensor* output_values = GetOutput(context, node, kOutputValues); in Prepare() local
168 TF_LITE_ENSURE_EQ(context, input->type, output_values->type); in Prepare()
178 TfLiteTensor* output_values = GetOutput(context, node, kOutputValues); in Prepare() local
180 SetTensorToDynamic(output_values); in Prepare()
186 TfLiteTensor* output_values = GetOutput(context, node, kOutputValues); in Eval() local
[all …]
|
| D | maximum_minimum_test.cc | 65 std::initializer_list<data_type> output_values) { in TestModel() argument
71 EXPECT_THAT(m.GetOutput<data_type>(), ElementsAreArray(output_values)); in TestModel()
79 std::initializer_list<float> output_values) { in TestModel() argument
86 ElementsAreArray(ArrayFloatNear(output_values))); in TestModel()
|
| /external/tensorflow/tensorflow/contrib/gan/python/features/python/ |
| D | random_tensor_pool_test.py | 102 output_values = tensor_pool(input_values, pool_size=3)
103 self.assertEqual(len(output_values), len(input_values))
104 for output_value in output_values:
109 outs = session.run(output_values, {
119 output_values = tensor_pool(input_values, pool_size=5)
120 print('stuff: ', output_values)
122 self.assertIsInstance(output_values, list)
123 self.assertEqual(3, len(output_values))
125 self.assertIsInstance(output_values[0], list)
126 self.assertEqual(3, len(output_values[0]))
[all …]
|
| D | random_tensor_pool_impl.py | 126 output_values = _to_list(control_flow_ops.cond(
131 for input_value, output_value in zip(input_values, output_values):
134 return nest.pack_sequence_as(original_input_values, output_values)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deserialize_sparse_variant_op.cc | 60 const Tensor* output_values; in Compute() local
70 &output_values)); in Compute()
72 context->set_output(1, *output_values); in Compute()
153 Tensor* output_values; in Compute() local
162 1, {static_cast<int64>(total_non_zeros)}, &output_values)); in Compute()
259 output_values->vec<T>().slice(values_start, values_sizes) = \ in Compute()
305 const Tensor** output_values) { in GetAndValidateSparseTensorIndicesAndValues() argument
334 *output_values = serialized_values.get<Tensor>(); in GetAndValidateSparseTensorIndicesAndValues()
335 if (*output_values == nullptr) { in GetAndValidateSparseTensorIndicesAndValues()
339 if (!TensorShapeUtils::IsVector((*output_values)->shape())) { in GetAndValidateSparseTensorIndicesAndValues()
[all …]
|
| D | deserialize_sparse_string_op.cc | 81 Tensor output_values; in Compute() local
87 &output_indices, &output_values, &output_shape)); in Compute()
89 context->set_output(1, output_values); in Compute()
104 Tensor output_values; in Compute() local
110 &output_values, &output_shape)); in Compute()
135 values.push_back(output_values); in Compute()
229 Tensor* output_indices, Tensor* output_values, Tensor* output_shape) { in GetAndValidateSparseTensor() argument
242 TF_RETURN_IF_ERROR(this->Deserialize(serialized_values, output_values)); in GetAndValidateSparseTensor()
243 if (!TensorShapeUtils::IsVector(output_values->shape())) { in GetAndValidateSparseTensor()
247 output_values->shape().DebugString()); in GetAndValidateSparseTensor()
[all …]
|
| D | sparse_tensors_map_ops.cc | 296 Tensor output_values = Tensor(DataTypeToEnum<T>::value, {num_entries}); in Compute() local
299 auto output_values_t = output_values.vec<T>(); in Compute()
310 SparseTensor::Create(output_indices, output_values, in Compute()
388 const Tensor& output_values = st.values(); in Compute() local
396 OP_REQUIRES(context, TensorShapeUtils::IsVector(output_values.shape()), in Compute()
400 output_values.shape().DebugString())); in Compute()
402 context, DataTypeToEnum<T>::value == output_values.dtype(), in Compute()
406 "].values.dtype() == ", DataTypeString(output_values.dtype()))); in Compute()
409 OP_REQUIRES(context, num_entries == output_values.dim_size(0), in Compute()
414 " vs. ", output_values.dim_size(0))); in Compute()
[all …]
|
| D | sparse_softmax_op.cc | 78 Tensor *output_values = nullptr; in Compute() local
80 &output_values)); in Compute()
81 typename TTypes<T>::Flat output_flat = output_values->flat<T>(); in Compute()
|
| D | sparse_dense_binary_op_shared.cc | 104 Tensor *output_values = nullptr; in Compute() local
108 ctx->allocate_output(0, TensorShape({nnz}), &output_values)); in Compute()
159 output_values->flat<T>().device(ctx->eigen_device<Device>()) = in Compute()
|
| D | sparse_fill_empty_rows_op.cc | 152 auto output_values = output_values_t->vec<T>(); in Compute() local
153 output_values.device(d) = output_values.constant(default_value); in Compute()
169 output_values(output_i) = values(i); in Compute()
|
| D | serialize_sparse_op.cc | 241 Tensor output_values = Tensor(DataTypeToEnum<T>::value, {num_entries}); in Compute() local
244 auto output_values_t = output_values.vec<T>(); in Compute()
256 context, this->Serialize(output_values, &serialized_sparse_t(b, 1))); in Compute()
|
| D | quantization_utils_test.cc | 50 auto output_values = o_tensor.flat<quint8>(); in TestRequantizeMany() local
56 output_values.data()); in TestRequantizeMany()
66 int v = output_values(value_index); in TestRequantizeMany()
92 auto output_values = o_tensor.flat<qint32>(); in TestRequantizeMany8To32Bit() local
97 output_values.data()); in TestRequantizeMany8To32Bit()
102 const qint32 v = output_values(value_index); in TestRequantizeMany8To32Bit()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | ops_test.py | 143 output_indices, output_values, output_shape = result
145 self.assertEqual(2, len(output_values))
148 self.assertEqual(output_values, result.output_values)
152 self.assertAllEqual([2, 7, 11], output_values[0])
153 self.assertAllEqual([3, 5], output_values[1])
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 576 std::vector<Tensor> output_values; in Split() local
579 output_values.reserve(num_split); in Split()
616 output_values.emplace_back(DataTypeToEnum<T>::v(), in Split()
620 output_values_t.emplace_back(output_values[i].vec<T>()); in Split()
645 Create(output_indices[i], output_values[i], output_shapes[i], &tensor); in Split()
695 Tensor output_values(DataTypeToEnum<T>::v(), TensorShape({count})); in Slice()
698 auto output_values_t = output_values.vec<T>(); in Slice()
726 return SparseTensor(output_indices, output_values, output_shape); in Slice()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 949 output_indices, output_values, output_shape = gen_sparse_ops.sparse_slice(
957 return sparse_tensor.SparseTensor(output_indices, output_values,
1867 (output_indices, output_values, empty_row_indicator,
1875 values=output_values,
2045 output_indices, output_values, output_shape = (
2052 return sparse_tensor.SparseTensor(output_indices, output_values, output_shape)
2117 output_indices, output_values, output_shape = (
2125 return sparse_tensor.SparseTensor(output_indices, output_values, output_shape)
2681 output_indices, output_values, output_shape = (
2693 return sparse_tensor.SparseTensor(output_indices, output_values, output_shape)
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | session_jni.cc | 148 std::unique_ptr<TF_Tensor* []> output_values(new TF_Tensor*[noutputs]); in Java_org_tensorflow_Session_run() local
177 static_cast<int>(ninputs), outputs.get(), output_values.get(), in Java_org_tensorflow_Session_run()
191 t[i] = reinterpret_cast<jlong>(output_values[i]); in Java_org_tensorflow_Session_run()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 289 ShapeHandle output_values = c->Vector(InferenceContext::kUnknownDim); in __anone6e195410c02() local
298 c->set_output(out_idx++, output_values); in __anone6e195410c02()
332 ShapeHandle output_values = c->Vector(InferenceContext::kUnknownDim); in __anone6e195410e02() local
336 c->set_output(1, output_values); in __anone6e195410e02()
584 ShapeHandle output_values = c->Vector(InferenceContext::kUnknownDim); in __anone6e195411602() local
591 c->set_output(1, output_values); in __anone6e195411602()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseFillEmptyRows.pbtxt | 30 name: "output_values"
53 `output_values`.
|
| D | api_def_SparseSparseMinimum.pbtxt | 47 name: "output_values"
|
| D | api_def_SparseSparseMaximum.pbtxt | 47 name: "output_values"
|
| D | api_def_SparseReorder.pbtxt | 30 name: "output_values"
|
| D | api_def_SparseSlice.pbtxt | 36 name: "output_values"
|
| D | api_def_SparseSplit.pbtxt | 31 name: "output_values"
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | model.py | 750 step_state, output_values = state_update_fn(
754 assert set(output_values.keys()) == set(outputs)
758 output_value = output_values[output_key]
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | estimator_test.py | 900 output_values = {}
904 output_values[v.tag] = v
905 self.assertTrue('MSE' in output_values)
906 self.assertTrue(output_values['MSE'].HasField('histo'))
931 output_values = {}
935 output_values[v.tag] = v
936 self.assertTrue('PMT' in output_values)
937 self.assertTrue(output_values['PMT'].HasField('tensor'))
|