| /external/tensorflow/tensorflow/lite/ |
| D | model_test.cc | 625 ASSERT_EQ(t1->sparsity->dim_metadata[0].array_segments, nullptr); in TEST()
630 TfLiteIntArray* array_segments = TfLiteIntArrayCreate(3); in TEST() local
631 array_segments->data[0] = 0; in TEST()
632 array_segments->data[1] = 2; in TEST()
633 array_segments->data[2] = 3; in TEST()
634 ASSERT_TRUE(TfLiteIntArrayEqual(t1->sparsity->dim_metadata[1].array_segments, in TEST()
635 array_segments)); in TEST()
636 TfLiteIntArrayFree(array_segments); in TEST()
648 ASSERT_EQ(t1->sparsity->dim_metadata[2].array_segments, nullptr); in TEST()
653 ASSERT_EQ(t1->sparsity->dim_metadata[3].array_segments, nullptr); in TEST()
|
| D | interpreter_builder.cc | 98 if (src->array_segments() == nullptr || src->array_indices() == nullptr) { in ParseSparseIndexVector()
104 status = Copy(src->array_segments_as_Int32Vector(), &tgt->array_segments); in ParseSparseIndexVector()
108 Copy(src->array_segments_as_Uint16Vector(), &tgt->array_segments); in ParseSparseIndexVector()
111 status = Copy(src->array_segments_as_Uint8Vector(), &tgt->array_segments); in ParseSparseIndexVector()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| D | fully_connected.h | 52 const int* w1_segments = sparsity.dim_metadata[1].array_segments->data; in FullyConnectedSparseWeight()
104 const int* w1_segments = sparsity.dim_metadata[1].array_segments->data; in FullyConnectedSparseWeight1x16Impl()
135 const int* w1_segments = sparsity.dim_metadata[1].array_segments->data; in FullyConnectedSparseWeight1x4Impl()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/utils/ |
| D | sparsity_format_converter.cc | 243 TfLiteIntArrayToVector(sparsity.dim_metadata[i].array_segments); in FormatConverter()
335 const auto& array_segments = dim_metadata_[metadata_idx]; in Populate() local
337 for (int i = array_segments[prev_idx]; i < array_segments[prev_idx + 1]; in Populate()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.cc | 154 TfLiteIntArrayFree(metadata.array_segments); in TfLiteSparsityFree()
155 metadata.array_segments = nullptr; in TfLiteSparsityFree()
|
| D | common_test.cc | 139 t.sparsity->dim_metadata[1].array_segments = TfLiteIntArrayCreate(2); in TEST()
|
| D | common.h | 367 TfLiteIntArray* array_segments; member
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.cc | 151 const auto* array_segments = param.dim_metadata[i].array_segments; in PyDictFromSparsityParam() local
155 PyArrayFromIntVector(array_segments->data, array_segments->size)); in PyDictFromSparsityParam()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fully_connected.cc | 64 sparsity->dim_metadata[1].array_segments->size - 1; in CreateLedgerTensor()
71 const auto* array_segments = sparsity->dim_metadata[1].array_segments; in PopulateLedgerData() local
75 for (int i = 0; i < array_segments->size - 1; i++) { in PopulateLedgerData()
76 int row_start = array_segments->data[i]; in PopulateLedgerData()
77 int row_end = array_segments->data[i + 1]; in PopulateLedgerData()
|
| D | test_util.h | 270 auto array_segments = in AddConstSparseInput() local
280 SparseIndexVector_Int32Vector, array_segments, in AddConstSparseInput()
342 auto array_segments =
352 SparseIndexVector_Int32Vector, array_segments,
|
| D | lstm.cc | 105 sparsity->dim_metadata[1].array_segments->size - 1; in make_ledger()
114 const auto* array_segments = sparsity->dim_metadata[1].array_segments; in copy_ledger() local
119 for (int i = 0; i < array_segments->size - 1; i++) { in copy_ledger()
120 int row_start = array_segments->data[i]; in copy_ledger()
121 int row_end = array_segments->data[i + 1]; in copy_ledger()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 2157 BufferOffset<void> array_segments; in BuildSparsityParameters() local
2165 array_segments = tflite::CreateUint8Vector( in BuildSparsityParameters()
2172 array_segments = tflite::CreateUint16Vector( in BuildSparsityParameters()
2177 array_segments = tflite::CreateInt32Vector( in BuildSparsityParameters()
2216 array_segments, indices_type, array_indices); in BuildSparsityParameters()
|
| D | flatbuffer_import.cc | 608 tensor.sparsity->dim_metadata[i]->array_segments)); in BuildSparseConstOp()
|
| /external/mesa3d/include/tensorflow/lite/core/c/ |
| D | common.h | 380 TfLiteIntArray* array_segments; member
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_v3a.fbs | 146 // - If format is DimensionType.SPARSE_CSR then we use array_segments and
147 // array_indices to encode that dimension. array_segments represents how
154 array_segments:SparseIndexVector;
|
| D | schema_v3b.fbs | 155 // - If format is DimensionType.SPARSE_CSR then we use array_segments and
156 // array_indices to encode that dimension. array_segments represents how
163 array_segments:SparseIndexVector;
|
| D | schema.fbs | 156 // - If format is DimensionType.SPARSE_CSR then we use array_segments and
157 // array_indices to encode that dimension. array_segments represents how
164 array_segments:SparseIndexVector;
|
| D | schema_generated.h | 4478 tflite::SparseIndexVectorUnion array_segments{};
4502 const void *array_segments() const {
4507 …arseIndexVector_Int32Vector ? static_cast<const tflite::Int32Vector *>(array_segments()) : nullptr;
4510 …seIndexVector_Uint16Vector ? static_cast<const tflite::Uint16Vector *>(array_segments()) : nullptr;
4513 …arseIndexVector_Uint8Vector ? static_cast<const tflite::Uint8Vector *>(array_segments()) : nullptr;
4537 VerifySparseIndexVector(verifier, array_segments(), array_segments_type()) &&
4585 void add_array_segments(flatbuffers::Offset<void> array_segments) {
4586 fbb_.AddOffset(DimensionMetadata::VT_ARRAY_SEGMENTS, array_segments);
4610 flatbuffers::Offset<void> array_segments = 0,
4615 builder_.add_array_segments(array_segments);
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | test_schema.fbs | 143 // - If format is DimensionType.SPARSE_CSR then we use array_segments and
144 // array_indices to encode that dimension. array_segments represents how
151 array_segments:SparseIndexVector;
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier.cc | 127 if (dim_metadata->array_segments() == nullptr) { in CheckArraySegments()
|
| D | verifier_test.cc | 758 tensor->sparsity->dim_metadata[1]->array_segments.AsUint8Vector()->values = { in TEST()
|
| /external/mesa3d/src/gallium/targets/teflon/ |
| D | tflite-schema-v2.15.0.fbs | 160 // - If format is DimensionType.SPARSE_CSR then we use array_segments and
161 // array_indices to encode that dimension. array_segments represents how
168 array_segments:SparseIndexVector;
|