| /external/tensorflow/tensorflow/lite/ |
| D | model_test.cc | 444 ASSERT_EQ(t1->sparsity->dim_metadata[0].array_indices, nullptr); in TEST()
456 TfLiteIntArray* array_indices = TfLiteIntArrayCreate(3); in TEST() local
457 array_indices->data[0] = 0; in TEST()
458 array_indices->data[1] = 1; in TEST()
459 array_indices->data[2] = 1; in TEST()
460 ASSERT_TRUE(TfLiteIntArrayEqual(t1->sparsity->dim_metadata[1].array_indices, in TEST()
461 array_indices)); in TEST()
462 TfLiteIntArrayFree(array_indices); in TEST()
467 ASSERT_EQ(t1->sparsity->dim_metadata[2].array_indices, nullptr); in TEST()
472 ASSERT_EQ(t1->sparsity->dim_metadata[3].array_indices, nullptr); in TEST()
|
| D | interpreter_builder.cc | 96 if (src->array_segments() == nullptr || src->array_indices() == nullptr) { in ParseSparseIndexVector()
119 return Copy(src->array_indices_as_Int32Vector(), &tgt->array_indices); in ParseSparseIndexVector()
121 return Copy(src->array_indices_as_Uint16Vector(), &tgt->array_indices); in ParseSparseIndexVector()
123 return Copy(src->array_indices_as_Uint8Vector(), &tgt->array_indices); in ParseSparseIndexVector()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 77 std::vector<llvm_ir::IrArray::Index> array_indices; in EmitIndexAndSetExitBasicBlock() local
133 array_indices.emplace_back(linear_index_base, shape_, b_); in EmitIndexAndSetExitBasicBlock()
139 array_indices.emplace_back(linear_index, shape_, b_); in EmitIndexAndSetExitBasicBlock()
155 return array_indices; in EmitIndexAndSetExitBasicBlock()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.c | 131 TfLiteIntArrayFree(metadata.array_indices); in TfLiteSparsityFree()
132 metadata.array_indices = NULL; in TfLiteSparsityFree()
|
| D | common_test.cc | 136 t.sparsity->dim_metadata[1].array_indices = TfLiteIntArrayCreate(3); in TEST()
|
| D | common.h | 357 TfLiteIntArray* array_indices; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| D | fully_connected.h | 50 const int* w1_indices = sparsity.dim_metadata[1].array_indices->data; in FullyConnectedSparseWeight()
97 const int* w1_indices = sparsity.dim_metadata[1].array_indices->data; in FullyConnectedSparseWeight1x4Impl()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter.cc | 237 TfLiteIntArrayToVector(sparsity.dim_metadata[i].array_indices); in FormatConverter()
296 const auto& array_indices = dim_metadata_[metadata_idx + 1]; in Populate() local
299 indices[level] = array_indices[i]; in Populate()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.cc | 126 const auto* array_indices = param.dim_metadata[i].array_indices; in PyDictFromSparsityParam() local
132 PyArrayFromIntVector(array_indices->data, array_indices->size)); in PyDictFromSparsityParam()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fully_connected.cc | 62 ledger_size->data[0] = sparsity->dim_metadata[1].array_indices->size + in CreateLedgerTensor()
71 const auto* array_indices = sparsity->dim_metadata[1].array_indices; in PopulateLedgerData() local
85 if (array_indices->data[j] > UINT8_MAX) { in PopulateLedgerData()
90 static_cast<uint8_t>(array_indices->data[j]); in PopulateLedgerData()
|
| D | test_util.h | 242 auto array_indices = in AddConstSparseInput() local
249 SparseIndexVector_Int32Vector, array_indices); in AddConstSparseInput()
314 auto array_indices =
321 SparseIndexVector_Int32Vector, array_indices);
584 .array_indices->size; in ExtractVector()
|
| D | lstm.cc | 103 ledger_size->data[0] = sparsity->dim_metadata[1].array_indices->size + in make_ledger()
114 const auto* array_indices = sparsity->dim_metadata[1].array_indices; in copy_ledger() local
129 if (array_indices->data[j] > UINT8_MAX) { in copy_ledger()
134 static_cast<uint8_t>(array_indices->data[j]); in copy_ledger()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier_test.cc | 632 tensor->sparsity->dim_metadata[1]->array_indices.AsUint8Vector()->values[1] = in TEST()
715 tensor->sparsity->dim_metadata[1]->array_indices.AsUint8Vector()->values = { in TEST()
|
| D | verifier.cc | 183 if (dim_metadata->array_indices() == nullptr) { in CheckArrayIndices()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 1903 BufferOffset<void> array_indices; in BuildSparsityParameters() local
1908 array_indices = tflite::CreateUint8Vector( in BuildSparsityParameters()
1915 array_indices = tflite::CreateUint16Vector( in BuildSparsityParameters()
1920 array_indices = tflite::CreateInt32Vector( in BuildSparsityParameters()
1927 array_segments, indices_type, array_indices); in BuildSparsityParameters()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | test_schema.fbs | 144 // array_indices to encode that dimension. array_segments represents how
146 // in the previous dimension. array_indices represents the index of the
152 array_indices:SparseIndexVector;
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_v3a.fbs | 147 // array_indices to encode that dimension. array_segments represents how
149 // in the previous dimension. array_indices represents the index of the
155 array_indices:SparseIndexVector;
|
| D | schema.fbs | 151 // array_indices to encode that dimension. array_segments represents how
153 // in the previous dimension. array_indices represents the index of the
159 array_indices:SparseIndexVector;
|
| D | schema_generated.h | 3450 tflite::SparseIndexVectorUnion array_indices;
3492 const void *array_indices() const {
3497 …parseIndexVector_Int32Vector ? static_cast<const tflite::Int32Vector *>(array_indices()) : nullptr;
3500 …rseIndexVector_Uint16Vector ? static_cast<const tflite::Uint16Vector *>(array_indices()) : nullptr;
3503 …parseIndexVector_Uint8Vector ? static_cast<const tflite::Uint8Vector *>(array_indices()) : nullptr;
3514 VerifySparseIndexVector(verifier, array_indices(), array_indices_type()) &&
3564 void add_array_indices(flatbuffers::Offset<void> array_indices) {
3565 fbb_.AddOffset(DimensionMetadata::VT_ARRAY_INDICES, array_indices);
3586 flatbuffers::Offset<void> array_indices = 0) {
3588 builder_.add_array_indices(array_indices);
[all …]
|