Home
last modified time | relevance | path

Searched defs:array_indices (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc77 std::vector<llvm_ir::IrArray::Index> array_indices; in EmitIndexAndSetExitBasicBlock() local
/external/tensorflow/tensorflow/lite/testdata/
Dsparse_tensor.json31 "array_indices": {"values": [0, 1, 1]} object
/external/tensorflow/tensorflow/lite/tools/optimize/sparsity/
Dformat_converter.cc296 const auto& array_indices = dim_metadata_[metadata_idx + 1]; in Populate() local
/external/tensorflow/tensorflow/lite/
Dmodel_test.cc456 TfLiteIntArray* array_indices = TfLiteIntArrayCreate(3); in TEST() local
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper.cc126 const auto* array_indices = param.dim_metadata[i].array_indices; in PyDictFromSparsityParam() local
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.h242 auto array_indices = in AddConstSparseInput() local
Dfully_connected.cc71 const auto* array_indices = sparsity->dim_metadata[1].array_indices; in PopulateLedgerData() local
Dlstm.cc114 const auto* array_indices = sparsity->dim_metadata[1].array_indices; in copy_ledger() local
/external/tensorflow/tensorflow/lite/c/
Dcommon.h357 TfLiteIntArray* array_indices; member
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc1903 BufferOffset<void> array_indices; in BuildSparsityParameters() local