| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | unbatch_dataset_op.cc | 56 shapes_.emplace_back(std::move(partial_dim_sizes)); in Dataset()
59 shapes_.emplace_back(); in Dataset()
76 return shapes_; in output_shapes()
119 shapes_(params.dataset->output_shapes().size()) {} in Iterator()
143 shapes_[i]); in GetNextInternal()
170 shapes_[i] = tensors_[i].shape(); in GetNextInternal()
171 shapes_[i].RemoveDim(0); in GetNextInternal()
236 shapes_[i] = tensors_[i].shape(); in RestoreInternal()
237 shapes_[i].RemoveDim(0); in RestoreInternal()
249 std::vector<TensorShape> shapes_ TF_GUARDED_BY(mu_);
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | outfeed_ops.h | 89 OP_REQUIRES_OK(ctx, ctx->GetAttr("shapes", &shapes_)); in TpuOutfeedDequeueTupleOp()
92 ctx, shapes_.size() == dtypes_.size(), in TpuOutfeedDequeueTupleOp()
96 for (int i = 0; i < shapes_.size(); i++) { in TpuOutfeedDequeueTupleOp()
99 TensorShapeToXLAShape(dtypes_[i], shapes_[i], &xla_shape)); in TpuOutfeedDequeueTupleOp()
109 for (int i = 0; i < shapes_.size(); ++i) { in DoWork()
111 TF_RETURN_IF_ERROR(ctx->allocate_output(i, shapes_[i], &output)); in DoWork()
123 std::vector<TensorShape> shapes_;
|
| 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::__anon820c6a2d0111::PrelinearizeTupleOp
472 OP_REQUIRES_OK(ctx, ctx->GetAttr("shapes", &shapes_)); in TpuInfeedEnqueueTupleOp()
[all …]
|
| D | infeed_ops.h | 58 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::__anoned9053ee0111::InfeedDequeueTupleOp
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | tensor_dataset_op.cc | 46 shapes_.reserve(tensors_.size()); in Dataset()
49 shapes_.emplace_back(t.shape().dim_sizes()); in Dataset()
68 return shapes_; in output_shapes()
184 std::vector<PartialTensorShape> shapes_; member in tensorflow::data::TensorDatasetOp::Dataset
|
| D | sparse_tensor_slice_dataset_op.cc | 38 shapes_({{-1, sparse_tensor.dims() - 1},
50 return shapes_; in output_shapes()
223 const std::vector<PartialTensorShape> shapes_; member in tensorflow::data::__anon0985c5ac0111::Dataset
|
| D | tensor_slice_dataset_op.cc | 60 shapes_.emplace_back(std::move(element_dim_sizes)); in Dataset()
207 std::vector<TensorShape> shapes_; member in tensorflow::data::TensorSliceDatasetOp::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_t>>& shapes() const { return shapes_; } in shapes()
901 std::vector<std::vector<int64_t>> shapes_; variable
|
| /external/tensorflow/tensorflow/core/data/ |
| D | snapshot_utils.cc | 383 shapes_(shapes), in Dataset()
389 return shapes_; in output_shapes()
525 const std::vector<PartialTensorShape> shapes_; member in tensorflow::data::snapshot_util::Reader::Dataset
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 2076 TF_RETURN_IF_ERROR(shapes_.Merge(s1, s2)); in Merge()
2096 ShapeHandle actual_shape = shapes_.GetMergedValue(shape); in AsTensorProperties()
2111 const auto& actual_shape = shapes_.GetMergedValue(s); in GetMergedShape()
2132 DisjointSet<shape_inference::ShapeHandle> shapes_; member in tensorflow::grappler::SymbolicShapeManager
|