Home
last modified time | relevance | path

Searched defs:dense_size (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/testdata/
Dsparse_tensor.json24 "dense_size": 2 number
35 "dense_size": 2 number
39 "dense_size": 2 number
/external/tensorflow/tensorflow/core/kernels/
Dsparse_to_dense_op_gpu.cu.cc55 __global__ void SetDefaultValue(const T default_value, const int64 dense_size, in SetDefaultValue()
111 Status LaunchComputeKernels(OpKernelContext* c, const int64 dense_size, in LaunchComputeKernels()
153 const int64 dense_size = dense->NumElements(); in operator ()() local
Dreshape_util.cc107 const int64_t dense_size = input_shape.num_elements(); in ReshapeSparseTensor() local
/external/tensorflow/tensorflow/lite/kernels/internal/utils/
Dsparsity_format_converter.cc221 const std::vector<int>& dense_size, in FormatConverter()
236 std::vector<int> dense_size(sparsity.dim_metadata_size); in FormatConverter() local
256 std::vector<TfLiteDimensionType> format, std::vector<int> dense_size, in InitSparseToDenseConverter()
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example_test.cc193 const char* text_def, int dense_size = 2) { in ParseExampleOpModel()
Dparse_example.cc656 int dense_size = 0; member
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc5043 const size_t dense_size = context->tensors[t].bytes / sizeof(float); in PrepareOpsToDelegate() local
5053 const size_t dense_size = in PrepareOpsToDelegate() local
5065 const size_t dense_size = in PrepareOpsToDelegate() local
/external/tensorflow/tensorflow/lite/c/
Dcommon.h366 int dense_size; member
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc5291 size_t dense_size; in DensifyAndDequantizeConstTensor() local