OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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