Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dexample_parsing_ops_test.cc164 std::vector<PartialTensorShape> dense_shapes; in ParseExample() local
227 std::vector<PartialTensorShape> dense_shapes; in ParseExampleV2() local
294 std::vector<PartialTensorShape> dense_shapes; in ParseSingleExample() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dexample_proto_helper.h197 std::vector<PartialTensorShape> dense_shapes; member
234 std::vector<PartialTensorShape> dense_shapes; member
H A Dexample_proto_helper.cc633 Status GetDenseShapes(const std::vector<PartialTensorShape>& dense_shapes, in GetDenseShapes()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/
H A Dparse_example.cc655 std::vector<tf::TensorShape> dense_shapes; member
720 auto dense_shapes = nodedef.attr().at("dense_shapes").list(); in PrepareParseExample() local
842 const auto& dense_shapes = data->dense_shapes; in EvalParseExample() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/fuzzing/
H A Dexample_proto_fast_parsing_fuzz.cc43 std::vector<PartialTensorShape> dense_shapes; in BuildGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dhandle_sparsetensors.cc60 mlir::RankedTensorType dense_shapes; member
/aosp_15_r20/external/tensorflow/tensorflow/core/example/
H A Dexample_parser_configuration.cc64 auto dense_shapes = attr_map.at("dense_shapes"); in ExtractExampleParserConfiguration() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dparsing_ops.cc32 Status AddDenseOutputShapes(const std::vector<TensorShapeType>& dense_shapes, in AddDenseOutputShapes()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dparsing_config.py396 dense_shapes=None, argument
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.h493 parallel_device::ParallelTensor* dense_shapes() const { in dense_shapes() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dparse_example_dataset_op.cc219 const std::vector<PartialTensorShape>& dense_shapes, in Dataset()