Home
last modified time | relevance | path

Searched defs:output_values (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dtopk_v2.cc58 TfLiteTensor* output_values = GetOutput(context, node, kOutputValues); in ResizeOutput() local
166 int32* output_indexes, T* output_values) { in TopK()
194 TfLiteTensor* output_values = GetOutput(context, node, kOutputValues); in Prepare() local
205 TfLiteTensor* output_values = GetOutput(context, node, kOutputValues); in Prepare() local
213 TfLiteTensor* output_values = GetOutput(context, node, kOutputValues); in Eval() local
Dmaximum_minimum_test.cc65 std::initializer_list<data_type> output_values) { in TestModel()
79 std::initializer_list<float> output_values) { in TestModel()
/external/tensorflow/tensorflow/core/kernels/
Ddeserialize_sparse_variant_op.cc60 const Tensor* output_values; in Compute() local
153 Tensor* output_values; in Compute() local
305 const Tensor** output_values) { in GetAndValidateSparseTensorIndicesAndValues()
Ddeserialize_sparse_string_op.cc81 Tensor output_values; in Compute() local
104 Tensor output_values; in Compute() local
229 Tensor* output_indices, Tensor* output_values, Tensor* output_shape) { in GetAndValidateSparseTensor()
Dsparse_softmax_op.cc74 Tensor *output_values = nullptr; in Compute() local
Dsparse_dense_binary_op_shared.cc104 Tensor *output_values = nullptr; in Compute() local
Dserialize_sparse_op.cc194 Tensor output_values = Tensor(DataTypeToEnum<T>::value, {num_entries}); in operator ()() local
294 Tensor& output_values = serialized_sparse_t(b, 1).emplace<Tensor>( in operator ()() local
Dsparse_tensors_map_ops.cc296 Tensor output_values = Tensor(DataTypeToEnum<T>::value, {num_entries}); in Compute() local
388 const Tensor& output_values = st.values(); in Compute() local
Dsparse_fill_empty_rows_op.cc157 auto output_values = output_values_t->vec<T>(); in Compute() local
Dquantization_utils_test.cc50 auto output_values = o_tensor.flat<quint8>(); in TestRequantizeMany() local
92 auto output_values = o_tensor.flat<qint32>(); in TestRequantizeMany8To32Bit() local
/external/tensorflow/tensorflow/core/ops/
Dsparse_ops.cc289 ShapeHandle output_values = c->Vector(InferenceContext::kUnknownDim); in __anon5983aeac0c02() local
332 ShapeHandle output_values = c->Vector(InferenceContext::kUnknownDim); in __anon5983aeac0e02() local
584 ShapeHandle output_values = c->Vector(InferenceContext::kUnknownDim); in __anon5983aeac1602() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_cholesky_op.cc178 Tensor output_values(cpu_allocator(), DataTypeToEnum<T>::value, in Compute() local
Dsparse_mat_mul_op.cc218 Tensor output_values(cpu_allocator(), DataTypeToEnum<T>::value, in Compute() local
/external/tensorflow/tensorflow/java/src/main/native/
Dsession_jni.cc148 std::unique_ptr<TF_Tensor* []> output_values(new TF_Tensor*[noutputs]); in Java_org_tensorflow_Session_run() local
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h462 std::vector<Tensor> output_values; in Split() local
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_test.cc2649 std::vector<Tensor> output_values; in FeedFetchBenchmarkHelper() local
2662 std::vector<Tensor> output_values; in FeedFetchBenchmarkHelper() local
2667 std::vector<Tensor> output_values; in FeedFetchBenchmarkHelper() local
/external/tensorflow/tensorflow/c/
Dc_api.cc2271 TF_Tensor** output_values, int noutputs, in TF_SessionRun()
2353 TF_Tensor** output_values, int noutputs, in TF_SessionPRun()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1461 auto& output_values = model->GetArray(op->outputs[0]); in ProcessTopkV2Operator() local