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.cc150 std::vector<llvm_ir::IrArray::Index> array_indices; in EmitLogicalIndexAndSetExitBasicBlock() local
240 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/kernels/internal/utils/
Dsparsity_format_converter.cc336 const auto& array_indices = dim_metadata_[metadata_idx + 1]; in Populate() local
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper.cc152 const auto* array_indices = param.dim_metadata[i].array_indices; in PyDictFromSparsityParam() local
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.h274 auto array_indices = in AddConstSparseInput() local
Dfully_connected.cc72 const auto* array_indices = sparsity->dim_metadata[1].array_indices; in PopulateLedgerData() local
Dlstm.cc115 const auto* array_indices = sparsity->dim_metadata[1].array_indices; in copy_ledger() local
/external/tensorflow/tensorflow/lite/
Dmodel_test.cc638 TfLiteIntArray* array_indices = TfLiteIntArrayCreate(3); in TEST() local
/external/tensorflow/tensorflow/lite/c/
Dcommon.h368 TfLiteIntArray* array_indices; member
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc2192 BufferOffset<void> array_indices; in BuildSparsityParameters() local