| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | unbatch_dataset_op.cc | 51 shapes_.emplace_back(std::move(partial_dim_sizes)); in Dataset()
54 shapes_.emplace_back(); in Dataset()
71 return shapes_; in output_shapes()
114 shapes_(params.dataset->output_shapes().size()) {} in Iterator()
136 shapes_[i]); in GetNextInternal()
163 shapes_[i] = tensors_[i].shape(); in GetNextInternal()
164 shapes_[i].RemoveDim(0); in GetNextInternal()
228 shapes_[i] = tensors_[i].shape(); in RestoreInternal()
229 shapes_[i].RemoveDim(0); in RestoreInternal()
241 std::vector<TensorShape> shapes_ TF_GUARDED_BY(mu_);
[all …]
|
| D | snapshot_util.cc | 366 shapes_(shapes), in Dataset()
372 return shapes_; in output_shapes()
478 const std::vector<PartialTensorShape> shapes_; member in tensorflow::data::snapshot_util::Reader::Dataset
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | infeed_ops.cc | 297 OP_REQUIRES_OK(ctx, ctx->GetAttr("shapes", &shapes_)); in PrelinearizeTupleOp()
300 ctx, shapes_.size() == dtypes_.size(), in PrelinearizeTupleOp()
303 shapes_.size(), ", dtypes length = ", dtypes_.size())); in PrelinearizeTupleOp()
306 for (int i = 0; i < shapes_.size(); i++) { in PrelinearizeTupleOp()
309 TensorShapeToXLAShape(dtypes_[i], shapes_[i], &xla_shape)); in PrelinearizeTupleOp()
321 OP_REQUIRES(ctx, values.size() == shapes_.size(), in Compute()
335 OP_REQUIRES(ctx, input_tensor.shape() == shapes_[i], in Compute()
338 "; expected ", shapes_[i].DebugString(), ", got ", in Compute()
368 std::vector<TensorShape> shapes_; member in tensorflow::__anon86ada7cf0111::PrelinearizeTupleOp
452 OP_REQUIRES_OK(ctx, ctx->GetAttr("shapes", &shapes_)); in TpuInfeedEnqueueTupleOp()
[all …]
|
| D | outfeed_ops.cc | 69 OP_REQUIRES_OK(ctx, ctx->GetAttr("shapes", &shapes_)); in TpuOutfeedDequeueTupleOp()
72 ctx, shapes_.size() == dtypes_.size(), in TpuOutfeedDequeueTupleOp()
76 for (int i = 0; i < shapes_.size(); i++) { in TpuOutfeedDequeueTupleOp()
79 TensorShapeToXLAShape(dtypes_[i], shapes_[i], &xla_shape)); in TpuOutfeedDequeueTupleOp()
92 for (int i = 0; i < shapes_.size(); ++i) { in DoWork()
94 TF_RETURN_IF_ERROR(ctx->allocate_output(i, shapes_[i], &output)); in DoWork()
|
| D | infeed_ops.h | 58 std::vector<TensorShape> shapes_;
|
| D | outfeed_ops.h | 59 std::vector<TensorShape> shapes_;
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | infeed_op.cc | 116 OP_REQUIRES_OK(ctx, ctx->GetAttr("shapes", &shapes_)); in InfeedDequeueTupleOp()
118 for (int i = 0; i < shapes_.size(); i++) { in InfeedDequeueTupleOp()
121 TensorShapeToXLAShape(dtypes_[i], shapes_[i], &xla_shape)); in InfeedDequeueTupleOp()
145 for (int i = 0; i < shapes_.size(); ++i) { in Compile()
151 std::vector<TensorShape> shapes_; member in tensorflow::__anondc5d46500111::InfeedDequeueTupleOp
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | tensor_dataset_op.cc | 42 shapes_.reserve(tensors_.size()); in Dataset()
45 shapes_.emplace_back(t.shape().dim_sizes()); in Dataset()
58 return shapes_; in output_shapes()
146 std::vector<PartialTensorShape> shapes_; member in tensorflow::data::TensorDatasetOp::Dataset
|
| D | tensor_slice_dataset_op.cc | 51 shapes_.emplace_back(std::move(element_dim_sizes)); in Dataset()
139 dataset()->shapes_[i]); in GetNextInternal()
171 std::vector<TensorShape> shapes_; member in tensorflow::data::TensorSliceDatasetOp::Dataset
|
| D | sparse_tensor_slice_dataset_op.cc | 38 shapes_({{-1, sparse_tensor.dims() - 1},
50 return shapes_; in output_shapes()
221 const std::vector<PartialTensorShape> shapes_; member in tensorflow::data::__anon1b965c8e0111::Dataset
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scoped_allocator_ops.cc | 32 OP_REQUIRES_OK(context, context->GetAttr("shapes", &shapes_)); in ScopedAllocatorOp()
41 ScopedAllocatorMgr::PopulateFields(id_, shapes_, dtype_, &fields_); in ScopedAllocatorOp()
76 std::vector<TensorShape> shapes_; member in tensorflow::ScopedAllocatorOp
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.h | 788 shapes_.clear(); in SnapshotNodes()
815 shapes_.push_back(node->shape()); in SnapshotNodes()
856 const std::vector<std::vector<int64>>& shapes() const { return shapes_; } in shapes()
901 std::vector<std::vector<int64>> shapes_; variable
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 2032 TF_RETURN_IF_ERROR(shapes_.Merge(s1, s2)); in Merge()
2052 ShapeHandle actual_shape = shapes_.GetMergedValue(shape); in AsTensorProperties()
2067 const auto& actual_shape = shapes_.GetMergedValue(s); in GetMergedShape()
2088 DisjointSet<shape_inference::ShapeHandle> shapes_; member in tensorflow::grappler::SymbolicShapeManager
|