| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter_test.cc | 26 const std::vector<int> dense_values = {6, 0, 9, 8, 0, 0, 0, 0, 5, 0, 0, 7}; in TEST() local
32 converter.DenseToSparse(dense_values.data()); in TEST()
46 EXPECT_EQ(data_back, dense_values); in TEST()
48 std::vector<int> dense_data(dense_values.size()); in TEST()
51 EXPECT_EQ(dense_data, dense_values); in TEST()
55 const std::vector<int> dense_values = {6, 0, 9, 8, 0, 0, 0, 0, 5, 0, 0, 7}; in TEST() local
61 converter.DenseToSparse(dense_values.data()); in TEST()
77 EXPECT_EQ(data_back, dense_values); in TEST()
79 std::vector<int> dense_data(dense_values.size()); in TEST()
82 EXPECT_EQ(dense_data, dense_values); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | densify_test.cc | 67 std::vector<float> dense_values = {6, 0, 9, 8, 0, 0, 0, 0, 5, 0, 0, 7}; in TEST() local
74 DensifyOpModel<float> m(input, dense_values); in TEST()
77 EXPECT_THAT(m.GetOutput(), ElementsAreArray(dense_values)); in TEST()
81 std::vector<float> dense_values = {6, 0, 9, 8, 0, 0, 0, 0, 5, 0, 0, 7}; in TEST() local
88 DensifyOpModel<float> m(input, dense_values); in TEST()
91 EXPECT_THAT(m.GetOutput(), ElementsAreArray(dense_values)); in TEST()
95 std::vector<int8_t> dense_values = {6, 0, 9, 8, 0, 0, 0, 0, 5, 0, 0, 7}; in TEST() local
102 DensifyOpModel<int8_t> m(input, dense_values); in TEST()
105 EXPECT_THAT(m.GetOutput(), ElementsAreArray(dense_values)); in TEST()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_RaggedTensorToVariantGradient.pbtxt | 19 Shape of the dense_values that was used as an input to the
26 Gradient for the dense_values of the RaggedTensorToVariant op.
33 Computes the gradient for the dense_values input to the RaggedTensorToVariant
|
| D | api_def_RaggedGather.pbtxt | 15 at the python level from dense_values to flat_values, so dense_values is the
|
| D | api_def_ParseSingleExample.pbtxt | 70 (D0, D1, ..., DN) then the shape of output Tensor dense_values[j]
72 ..., DN), the shape of the output Tensor dense_values[j] will be (M,
|
| D | api_def_ParseExample.pbtxt | 66 Tensor dense_values[j] will be (|serialized|, D0, D1, ..., DN):
69 the shape of the output Tensor dense_values[j] will be
|
| D | api_def_ParseExampleV2.pbtxt | 97 Tensor dense_values[j] will be (|serialized|, D0, D1, ..., DN):
100 the shape of the output Tensor dense_values[j] will be
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 64 std::vector<TfLiteTensor*> dense_values; member
346 TfLiteTensor* out = result->dense_values[d]; in FastParseSerializedExample()
472 hasher, &result->dense_values, &varlen_dense_buffers, &sparse_buffers, in FastParseExampleLite()
561 TfLiteTensor* values = result->dense_values[d]; in FastParseExampleLite()
594 if (result->dense_values[d]->type == kTfLiteString) { in FastParseExampleLite()
597 const int batch_size = result->dense_values[d]->dims->data[0]; in FastParseExampleLite()
618 reinterpret_cast<char*>(result->dense_values[d]->data.raw); in FastParseExampleLite()
619 if (result->dense_values[d]->bytes < required_bytes) { in FastParseExampleLite()
620 if (result->dense_values[d]->data.raw) { in FastParseExampleLite()
621 free(result->dense_values[d]->data.raw); in FastParseExampleLite()
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 861 dense_values, dense_indices, sparse_values, sparse_indices = (
863 if dense_values:
864 dense_results = self._do_batch_all_reduce(reduce_op, dense_values)
875 def _do_batch_all_reduce(self, reduce_op, dense_values): argument
880 (len(dense_values), self._all_reduce_alg, self._num_packs), 10)
882 destinations = dense_values[0]._devices # pylint: disable=protected-access
883 grouped = _group_value_by_device(dense_values)
904 return _ungroup_and_make_mirrored(reduced, dense_values[0], reduce_op)
1183 dense_values, dense_indices, sparse_values, sparse_indices = (
1185 if dense_values:
[all …]
|
| D | cross_device_utils.py | 621 dense_values = []
630 dense_values.append(v)
632 return dense_values, dense_indices, sparse_values, sparse_indices
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_conversion_ops.py | 175 dense_values = op.inputs[-1]
181 dense_values_shape=array_ops.shape(dense_values),
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_range_tflite.cc | 160 TfLiteTensor& dense_values = in Prepare() local
162 SetTensorToDynamic(&dense_values); in Prepare()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops.cc | 257 OpOutputList dense_values; in WriteOutput() local
261 TF_RETURN_IF_ERROR(ctx->output_list("dense_values", &dense_values)); in WriteOutput()
266 dense_values.set(d, result.dense_values[d]); in WriteOutput()
369 OpOutputList dense_values; in Compute() local
373 OP_REQUIRES_OK(ctx, ctx->output_list("dense_values", &dense_values)); in Compute()
378 dense_values.set(d, result.dense_values[d]); in Compute()
720 context_dense_values.set(d, context_result.dense_values[d]); in WriteOutput()
723 feature_list_dense_values.set(d, feature_list_result.dense_values[d]); in WriteOutput()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1242 result->dense_values.reserve(config.dense.size()); in FastParseExample()
1247 result->dense_values.push_back(std::move(fixed_dense_values[d])); in FastParseExample()
1384 result->dense_values[d] = values; in FastParseExample()
1475 result->dense_values.reserve(config.dense.size()); in FastParseSingleExample()
1487 result->dense_values.emplace_back(config.dense[d].dtype, values_shape); in FastParseSingleExample()
1490 result->dense_values.emplace_back(); in FastParseSingleExample()
1586 Tensor* out = &result->dense_values[d]; in FastParseSingleExample()
1717 out = &result->dense_values[d]; in FastParseSingleExample()
1781 result->dense_values[d] = config.dense[d].default_value; in FastParseSingleExample()
1789 result->dense_values[d] = Tensor(config.dense[d].dtype, empty_shape); in FastParseSingleExample()
[all …]
|
| D | example_proto_fast_parsing.h | 122 std::vector<Tensor> dense_values; member
|
| D | example_proto_helper.h | 81 std::vector<Tensor*>* dense_values,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ParseSingleExample.pbtxt | 26 name: "dense_values"
|
| D | ParseExample.pbtxt | 40 name: "dense_values"
|
| D | ParseExampleV2.pbtxt | 42 name: "dense_values"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ParseSingleExample.pbtxt | 26 name: "dense_values"
|
| D | ParseExample.pbtxt | 40 name: "dense_values"
|
| D | ParseExampleV2.pbtxt | 42 name: "dense_values"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_ops.py | 363 (sparse_indices, sparse_values, sparse_shapes, dense_values,
375 sparse_tensors + dense_values + ragged_tensors))
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 582 auto dense_values = mlir::DenseIntElementsAttr::get(type, value); in Create1DConstantOp() local
583 return builder->create<TF::ConstOp>(loc, dense_values); in Create1DConstantOp()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parse_example_dataset_op.cc | 663 TF_RETURN_IF_ERROR(CheckOutputTensor(example_result.dense_values[d], in ParseExample()
665 (*output)[output_index] = example_result.dense_values[d]; in ParseExample()
|