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::__anoned9053ee0111::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.cc253 std::vector<PartialTensorShape> shapes_; member in tensorflow::data::experimental::__anon3dae43170111::UnbatchDatasetOp::Dataset
/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_dataset_op.cc184 std::vector<PartialTensorShape> shapes_; member in tensorflow::data::TensorDatasetOp::Dataset
Dsparse_tensor_slice_dataset_op.cc223 const std::vector<PartialTensorShape> shapes_; member in tensorflow::data::__anon0985c5ac0111::Dataset
Dtensor_slice_dataset_op.cc207 std::vector<TensorShape> shapes_; member in tensorflow::data::TensorSliceDatasetOp::Dataset
/external/tensorflow/tensorflow/core/tpu/kernels/
Dinfeed_ops.cc368 std::vector<TensorShape> shapes_; member in tensorflow::__anon820c6a2d0111::PrelinearizeTupleOp
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h901 std::vector<std::vector<int64_t>> 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.cc2132 DisjointSet<shape_inference::ShapeHandle> shapes_; member in tensorflow::grappler::SymbolicShapeManager