Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/testdata/
H A Dsparse_tensor.json31 "array_indices": {"values": [0, 1, 1]} object
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/utils/
H A Dsparsity_format_converter.cc336 const auto& array_indices = dim_metadata_[metadata_idx + 1]; in Populate() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
H A Dinterpreter_wrapper.cc152 const auto* array_indices = param.dim_metadata[i].array_indices; in PyDictFromSparsityParam() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dtest_util.h274 auto array_indices = in AddConstSparseInput() local
H A Dfully_connected.cc72 const auto* array_indices = sparsity->dim_metadata[1].array_indices; in PopulateLedgerData() local
H A Dlstm.cc115 const auto* array_indices = sparsity->dim_metadata[1].array_indices; in copy_ledger() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dmodel_test.cc638 TfLiteIntArray* array_indices = TfLiteIntArrayCreate(3); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dcommon.h368 TfLiteIntArray* array_indices; member
/aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/
H A Dcommon.h381 TfLiteIntArray* array_indices; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_export.cc2192 BufferOffset<void> array_indices; in BuildSparsityParameters() local