| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | profiler_kineto.h | 75 std::vector<std::vector<int64_t>> shapes_; member
|
| H A D | profiler_legacy.h | 255 std::vector<std::vector<int64_t>> shapes_; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| H A D | infeed_op.cc | 151 std::vector<TensorShape> shapes_; member in tensorflow::__anon2b38e49f0111::InfeedDequeueTupleOp
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | ir.h | 177 std::vector<Shape> shapes_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | scoped_allocator_ops.cc | 76 std::vector<TensorShape> shapes_; member in tensorflow::ScopedAllocatorOp
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | unbatch_dataset_op.cc | 253 std::vector<PartialTensorShape> shapes_; member in tensorflow::data::experimental::__anonfb83e4280111::UnbatchDatasetOp::Dataset
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | tensor_dataset_op.cc | 184 std::vector<PartialTensorShape> shapes_; member in tensorflow::data::TensorDatasetOp::Dataset
|
| H A D | sparse_tensor_slice_dataset_op.cc | 223 const std::vector<PartialTensorShape> shapes_; member in tensorflow::data::__anonc1b3089d0111::Dataset
|
| H A D | tensor_slice_dataset_op.cc | 207 std::vector<TensorShape> shapes_; member in tensorflow::data::TensorSliceDatasetOp::Dataset
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | infeed_ops.cc | 368 std::vector<TensorShape> shapes_; member in tensorflow::__anon368a507e0111::PrelinearizeTupleOp
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/internal/ |
| H A D | tfprof_node.h | 901 std::vector<std::vector<int64_t>> shapes_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | snapshot_utils.cc | 525 const std::vector<PartialTensorShape> shapes_; member in tensorflow::data::snapshot_util::Reader::Dataset
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | graph_properties.cc | 2132 DisjointSet<shape_inference::ShapeHandle> shapes_; member in tensorflow::grappler::SymbolicShapeManager
|