Home
last modified time | relevance | path

Searched defs:shapes_ (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dinfeed_op.cc151 std::vector<TensorShape> shapes_; member in tensorflow::__anonb4f794950111::InfeedDequeueTupleOp
/external/tensorflow/tensorflow/core/kernels/
Dscoped_allocator_ops.cc76 std::vector<TensorShape> shapes_; member in tensorflow::ScopedAllocatorOp
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunbatch_dataset_op.cc252 std::vector<PartialTensorShape> shapes_; member in tensorflow::data::experimental::__anon46dcc45e0111::UnbatchDatasetOp::Dataset
/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_dataset_op.cc173 std::vector<PartialTensorShape> shapes_; member in tensorflow::data::TensorDatasetOp::Dataset
Dtensor_slice_dataset_op.cc175 std::vector<TensorShape> shapes_; member in tensorflow::data::TensorSliceDatasetOp::Dataset
Dsparse_tensor_slice_dataset_op.cc221 const std::vector<PartialTensorShape> shapes_; member in tensorflow::data::__anonacb5ba130111::Dataset
/external/tensorflow/tensorflow/core/tpu/kernels/
Dinfeed_ops.cc368 std::vector<TensorShape> shapes_; member in tensorflow::__anon241814340111::PrelinearizeTupleOp
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h901 std::vector<std::vector<int64>> shapes_; variable
/external/tensorflow/tensorflow/core/data/
Dsnapshot_utils.cc525 const std::vector<PartialTensorShape> shapes_; member in tensorflow::data::snapshot_util::Reader::Dataset
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc2120 DisjointSet<shape_inference::ShapeHandle> shapes_; member in tensorflow::grappler::SymbolicShapeManager