HomeSort by relevance Sort by last modified time
    Searched refs:shapes_ (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/core/kernels/data/experimental/
unbatch_dataset_op.cc 49 shapes_.emplace_back(std::move(partial_dim_sizes));
52 shapes_.emplace_back();
69 return shapes_;
91 shapes_(params.dataset->output_shapes().size()) {}
112 shapes_[i]);
139 shapes_[i] = tensors_[i].shape();
140 shapes_[i].RemoveDim(0);
203 shapes_[i] = tensors_[i].shape();
204 shapes_[i].RemoveDim(0);
216 std::vector<TensorShape> shapes_ GUARDED_BY(mu_)
220 std::vector<PartialTensorShape> shapes_; member in class:tensorflow::data::__anon45296::UnbatchDatasetOp::Dataset
    [all...]
  /external/tensorflow/tensorflow/core/kernels/data/
tensor_dataset_op.cc 53 shapes_.emplace_back(t.shape().dim_sizes());
65 return shapes_;
144 std::vector<PartialTensorShape> shapes_; member in class:tensorflow::data::__anon45337::TensorDatasetOp::Dataset
tensor_slice_dataset_op.cc 76 shapes_.emplace_back(std::move(partial_dim_sizes));
88 return shapes_;
140 TensorShape(dataset()->shapes_[i].dim_sizes()));
179 std::vector<PartialTensorShape> shapes_; member in class:tensorflow::data::__anon45339::TensorSliceDatasetOp::Dataset
sparse_tensor_slice_dataset_op.cc 38 shapes_({{-1, sparse_tensor.dims() - 1},
50 return shapes_;
214 const std::vector<PartialTensorShape> shapes_; member in class:tensorflow::data::__anon45333::Dataset
  /external/tensorflow/tensorflow/contrib/ignite/kernels/dataset/
ignite_dataset.cc 44 shapes_(shapes) {
69 return shapes_;
ignite_dataset.h 58 const std::vector<PartialTensorShape> shapes_; member in class:tensorflow::IgniteDataset
  /external/tensorflow/tensorflow/core/kernels/
scoped_allocator_ops.cc 32 OP_REQUIRES_OK(context, context->GetAttr("shapes", &shapes_));
41 ScopedAllocatorMgr::PopulateFields(id_, shapes_, dtype_, &fields_);
75 std::vector<TensorShape> shapes_; member in class:tensorflow::ScopedAllocatorOp
  /external/tensorflow/tensorflow/core/profiler/internal/
tfprof_node.h 789 shapes_.clear();
816 shapes_.push_back(node->shape());
857 const std::vector<std::vector<int64>>& shapes() const { return shapes_; }
902 std::vector<std::vector<int64>> shapes_; member in class:tensorflow::tfprof::TFMultiGraphNode
  /external/tensorflow/tensorflow/core/grappler/costs/
graph_properties.cc 1754 DisjointSet<shape_inference::ShapeHandle> shapes_; member in class:tensorflow::grappler::SymbolicShapeManager
    [all...]

Completed in 767 milliseconds