Home
last modified time | relevance | path

Searched defs:num_sparse (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dparsing_ops_test.cc90 int unknown_outer_dims = 0) { in TEST()
307 int unknown_outer_dims = 0) { in TEST()
H A Dragged_array_ops.cc84 int num_sparse; in __anon1fd7ceb10102() local
H A Dparsing_ops.cc45 void AddSparseOutputShapes(int num_sparse, const ShapeHandle input_shape, in AddSparseOutputShapes()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dexample_proto_helper.h190 int64_t num_sparse; member
217 int num_sparse; in Init() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dragged_cross_op.cc300 int num_sparse; in RaggedCrossOp() local
385 const auto num_sparse = sparse_indices_list.size(); in ValidateInput() local
H A Dexample_parsing_ops_test.cc260 int num_sparse = opt.benchmark_type == kSparse ? num_keys : 0; in ParseExampleV2() local
/aosp_15_r20/external/tensorflow/tensorflow/core/example/
H A Dexample_parser_configuration.cc58 auto num_sparse = attr_map.at("Nsparse").i(); in ExtractExampleParserConfiguration() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/
H A Dparse_example.cc733 int num_sparse = 0; in PrepareParseExample() local