Home
last modified time | relevance | path

Searched refs:TensorSlice (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dtensor_slice_test.cc30 TensorSlice s(3); in TEST()
44 TensorSlice s({{0, -1}, {0, 10}, {14, 1}, {0, -1}}); in TEST()
62 TensorSlice s(proto); in TEST()
69 TensorSlice s = TensorSlice::ParseOrDie("-:-:1,3:4,5"); in TEST()
83 TensorSlice slice; in TEST()
84 Status s = TensorSlice::Parse("-:-:1,3:4:5", &slice); in TEST()
92 TensorSlice slice; in TEST()
93 Status s = TensorSlice::Parse("-:-1,3", &slice); in TEST()
103 TensorSlice s = in TEST()
104 TensorSlice::ParseOrDie("9223372036854775807,9223372036854775807"); in TEST()
[all …]
Dtensor_slice.cc26 TensorSlice::TensorSlice(int dim) { SetFullSlice(dim); } in TensorSlice() function in tensorflow::TensorSlice
28 TensorSlice::TensorSlice(const TensorSliceProto& proto) { in TensorSlice() function in tensorflow::TensorSlice
37 TensorSlice::TensorSlice( in TensorSlice() function in tensorflow::TensorSlice
47 Status TensorSlice::Parse(const string& str, TensorSlice* slice) { in Parse()
80 void TensorSlice::Clear() { in Clear()
85 bool TensorSlice::IsFull() const { in IsFull()
92 void TensorSlice::SetFullSlice(int dim) { in SetFullSlice()
102 void TensorSlice::Extend(int dim) { in Extend()
113 void TensorSlice::AsProto(TensorSliceProto* proto) const { in AsProto()
124 string TensorSlice::DebugString() const { in DebugString()
[all …]
Dtensor_slice.h34 class TensorSlice {
45 TensorSlice() {} in TensorSlice() function
46 explicit TensorSlice(int dim);
47 explicit TensorSlice(const TensorSliceProto& proto);
48 explicit TensorSlice(std::initializer_list<std::pair<int64, int64>> extents);
50 static Status Parse(const string& str, TensorSlice* output);
51 static TensorSlice ParseOrDie(const string& str) { in ParseOrDie()
52 TensorSlice ret; in ParseOrDie()
131 bool Intersect(const TensorSlice& other, TensorSlice* result) const;
133 bool Overlaps(const TensorSlice& other) const { in Overlaps()
[all …]
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_set_test.cc51 TensorSlice slice_1 = TensorSlice::ParseOrDie("0,2:-"); in TEST()
59 TensorSlice slice_2 = TensorSlice::ParseOrDie("2,2:0,3"); in TEST()
67 TensorSlice slice_3 = TensorSlice::ParseOrDie("3,1:3,2"); in TEST()
85 TensorSlice s = TensorSlice::ParseOrDie("0,2:-"); in TEST()
86 std::vector<std::pair<TensorSlice, string>> results; in TEST()
100 TensorSlice s = TensorSlice::ParseOrDie("1,1:-"); in TEST()
101 std::vector<std::pair<TensorSlice, string>> results; in TEST()
116 TensorSlice s = TensorSlice::ParseOrDie("1,2:0,3"); in TEST()
117 std::vector<std::pair<TensorSlice, string>> results; in TEST()
140 TensorSlice s = TensorSlice::ParseOrDie("1,2:2,3"); in TEST()
[all …]
Dtensor_slice_util_test.cc46 TensorSlice slice_s = TensorSlice::ParseOrDie("1,2:1,3"); in TEST()
47 TensorSlice slice_d = TensorSlice::ParseOrDie("1,2:1,3"); in TEST()
62 TensorSlice slice_s = TensorSlice::ParseOrDie("1,2:1,3"); in TEST()
63 TensorSlice slice_d = TensorSlice::ParseOrDie("3,1:2,3"); in TEST()
89 TensorSlice slice_s = TensorSlice::ParseOrDie("0,3:0,3"); in TEST()
90 TensorSlice slice_d = TensorSlice::ParseOrDie("1,2:1,4"); in TEST()
Dtensor_slice_reader_test.cc72 TensorSlice slice = TensorSlice::ParseOrDie("0,2:-"); in SimpleFloatHelper()
96 TensorSlice slice = TensorSlice::ParseOrDie("2,2:0,3"); in SimpleFloatHelper()
102 TensorSlice slice = TensorSlice::ParseOrDie("3,1:3,2"); in SimpleFloatHelper()
138 TensorSlice s = TensorSlice::ParseOrDie("0,2:-"); in SimpleFloatHelper()
153 TensorSlice s = TensorSlice::ParseOrDie("1,1:-"); in SimpleFloatHelper()
168 TensorSlice s = TensorSlice::ParseOrDie("1,2:2,3"); in SimpleFloatHelper()
197 TensorSlice slice = TensorSlice::ParseOrDie("0,2:-"); in SimpleIntXHelper()
221 TensorSlice slice = TensorSlice::ParseOrDie("2,2:0,3"); in SimpleIntXHelper()
227 TensorSlice slice = TensorSlice::ParseOrDie("3,1:3,2"); in SimpleIntXHelper()
263 TensorSlice s = TensorSlice::ParseOrDie("0,2:-"); in SimpleIntXHelper()
[all …]
Dtensor_slice_writer_test.cc41 const TensorSlice& slice, SavedSlice* ss);
81 TensorSlice slice = TensorSlice::ParseOrDie("-:0,1"); in TEST()
89 TensorSlice slice = TensorSlice::ParseOrDie("-:3,1"); in TEST()
98 TensorSlice slice = TensorSlice::ParseOrDie("-:-"); in TEST()
106 TensorSlice slice = TensorSlice::ParseOrDie("-:3,1"); in TEST()
114 TensorSlice slice = TensorSlice::ParseOrDie("-:3,1"); in TEST()
129 const TensorSlice& slice, in GetData()
139 TensorSlice slice2(ss->slice()); in GetData()
177 TensorSlice s0(ssm.slice(0)); in CheckEntries()
178 TensorSlice s1(ssm.slice(1)); in CheckEntries()
[all …]
Dtensor_slice_set.h49 Status Register(const TensorSlice& slice, const string& tag);
55 const TensorSlice& slice,
56 std::vector<std::pair<tensorflow::TensorSlice, string>>* results) const;
59 TensorSlice slice;
77 TensorSlice slices_hull_;
86 const string& tag, const TensorSlice& slice,
Dtensor_slice_reader.h91 bool CopySliceData(const string& name, const TensorSlice& slice,
123 const string& name, const TensorSlice& slice,
124 std::vector<std::pair<TensorSlice, string>>* details) const;
146 const TensorSlice& slice, T* data) const { in CopySliceData()
147 std::vector<std::pair<TensorSlice, string>> details; in CopySliceData()
166 const TensorSlice& slice_s = x.first; in CopySliceData()
Dtensor_slice_set.cc33 Status TensorSliceSet::Register(const TensorSlice& slice, const string& tag) { in Register()
61 const TensorSlice& slice, in QueryMeta()
62 std::vector<std::pair<TensorSlice, string>>* results) const { in QueryMeta() argument
88 TensorSlice intersection; in QueryMeta()
114 const string& tag, const TensorSlice& slice, in RegisterTensorSlice()
Dsaved_tensor_slice_util_test.cc33 TensorSlice s = TensorSlice::ParseOrDie("-:-:1,3:4,5"); in TEST()
Dsaved_tensor_slice_util.cc28 string EncodeTensorNameSlice(const string& name, const TensorSlice& slice) { in EncodeTensorNameSlice()
46 tensorflow::TensorSlice* slice) { in DecodeTensorNameSlice()
91 TensorSlice* slice, TensorShape* shape_slice) { in ParseShapeAndSlice()
Dtensor_slice_util.h138 const TensorSlice& slice_s,
139 const TensorSlice& slice_d,
146 TensorSlice inter;
168 TensorSlice rel_s, rel_d;
Dtensor_slice_reader.cc173 TensorSlice ss_slice(tsp); in LoadShard()
190 const string& name, const TensorSlice& slice, in FindTensorSlice()
191 std::vector<std::pair<TensorSlice, string>>* details) const { in FindTensorSlice() argument
233 TensorSlice slice; in GetTensor()
Dsaved_tensor_slice_util.h47 const tensorflow::TensorSlice& slice);
51 tensorflow::TensorSlice* slice);
57 TensorSlice* slice, TensorShape* shape_slice);
Dtensor_slice_writer.h60 const TensorSlice& slice, const T* data);
99 const TensorSlice& slice, const T* data) { in Add()
/external/tensorflow/tensorflow/core/kernels/
Dsave_op_test.cc149 TensorSlice s = TensorSlice::ParseOrDie("-"); in TEST_F()
168 TensorSlice s = TensorSlice::ParseOrDie("-"); in TEST_F()
187 TensorSlice s = TensorSlice::ParseOrDie("-:-"); in TEST_F()
206 TensorSlice s = TensorSlice::ParseOrDie("-:-"); in TEST_F()
225 TensorSlice s = TensorSlice::ParseOrDie("-:-"); in TEST_F()
243 TensorSlice s = TensorSlice::ParseOrDie("-:-"); in TEST_F()
261 TensorSlice s = TensorSlice::ParseOrDie("-"); in TEST_F()
279 TensorSlice s = TensorSlice::ParseOrDie("-"); in TEST_F()
297 TensorSlice s = TensorSlice::ParseOrDie("-"); in TEST_F()
315 TensorSlice s = TensorSlice::ParseOrDie("-"); in TEST_F()
[all …]
Dload_and_remap_matrix_op.cc168 std::vector<TensorSlice> tensor_slices; in Compute()
169 TensorSlice slice(tensor_shape.dims()); in Compute()
202 for (const TensorSlice& tensor_slice : tensor_slices) { in Compute()
Dsave_restore_tensor.cc104 TensorSlice slice(input.dims()); in SaveTensors()
193 TensorSlice slice_to_load(saved_shape.dims()); in RestoreTensor()
285 TensorSlice parsed_slice; in run()
Dpooling_ops_3d.cc238 TensorSlice src{{0, -1}, {c, 1}, {r, 1}, {p, 1}, {0, -1}}; in launch()
239 TensorSlice dst{{0, -1}, in launch()
413 TensorSlice src{{0, -1}, {c, 1}, {r, 1}, {p, 1}, {0, -1}}; in launch()
414 TensorSlice dst{{0, -1}, in launch()
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.h141 const TensorSlice& slice_spec, const Tensor& slice_tensor);
237 Status LookupTensorSlices(StringPiece key, std::vector<TensorSlice>* slices)
244 Status LookupSlice(StringPiece full_tensor_key, const TensorSlice& slice_spec,
283 const TensorSlice& slice_spec,
Dtensor_bundle_test.cc566 TensorSlice slice1 = TensorSlice::ParseOrDie("-:0,1"); in TEST()
567 TensorSlice slice2 = TensorSlice::ParseOrDie("-:1,9"); in TEST()
599 std::vector<TensorSlice> slices; in TEST()
612 const TensorSlice distinct_slice = TensorSlice::ParseOrDie("-:0,2"); in TEST()
630 const TensorSlice distinct_slice = TensorSlice::ParseOrDie("-:2,2"); in TEST()
644 TensorSlice::ParseOrDie("-:-"), kExpected)); in TEST()
646 TensorSlice::ParseOrDie("0,5:0,10"), in TEST()
654 const TensorSlice slice = TensorSlice::ParseOrDie("-:-"); in TEST()
663 const TensorSlice slice = TensorSlice::ParseOrDie("0,5:0,10"); in TEST()
672 const TensorSlice slice = TensorSlice::ParseOrDie("0,5:0,10"); in TEST()
[all …]
Dtensor_bundle.cc352 bool IsFullSlice(const TensorSlice& slice_spec, in IsFullSlice()
478 const TensorSlice& slice_spec, in AddSlice()
949 /* a full slice */ TensorSlice(TensorShape(entry.shape()).dims()), val); in Lookup()
967 /* a full slice */ TensorSlice(TensorShape(entry.shape()).dims()), val); in ReadCurrent()
972 std::vector<TensorSlice>* slices) { in LookupTensorSlices()
984 const TensorSlice& slice_spec, Tensor* val) { in LookupSlice()
993 const TensorSlice& slice_spec, Tensor* val) { in GetSliceValue()
999 std::vector<std::pair<TensorSlice, string>> details; in GetSliceValue()
1013 /* full slice */ TensorSlice(full_shape.dims()), &tensor_slices_)); in GetSliceValue()
1018 /* tag */ "", TensorSlice(slice), &tensor_slices_)); in GetSliceValue()
[all …]
/external/tensorflow/tensorflow/core/ops/
Dio_ops.cc116 TensorSlice parsed_slice; in __anon52e493120302()
230 TensorSlice parsed_slice; in __anon52e493120802()
/external/tensorflow/tensorflow/c/
Dcheckpoint_reader.cc128 TensorSlice(slice_proto))) in BuildV2VarMaps()

12