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.cc53 TensorSlice slice_1 = TensorSlice::ParseOrDie("0,2:-"); in TEST()
62 TensorSlice slice_2 = TensorSlice::ParseOrDie("2,2:0,3"); in TEST()
71 TensorSlice slice_3 = TensorSlice::ParseOrDie("3,1:3,2"); in TEST()
88 TensorSlice s = TensorSlice::ParseOrDie("0,2:-"); in TEST()
103 TensorSlice s = TensorSlice::ParseOrDie("1,1:-"); in TEST()
119 TensorSlice s = TensorSlice::ParseOrDie("1,2:0,3"); in TEST()
134 TensorSlice s = TensorSlice::ParseOrDie("1,2:2,3"); in TEST()
152 TensorSlice slice_1 = TensorSlice::ParseOrDie("0,2:-"); in TEST()
160 TensorSlice slice_2 = TensorSlice::ParseOrDie("2,2:0,3"); in TEST()
168 TensorSlice slice_3 = TensorSlice::ParseOrDie("3,1:3,2"); in TEST()
[all …]
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_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_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.h55 Status Register(const TensorSlice& slice, const string& tag,
63 bool Query(const TensorSlice& slice, float* data) const;
69 const TensorSlice& slice,
70 std::vector<std::pair<tensorflow::TensorSlice, string>>* results) const;
73 TensorSlice slice;
92 TensorSlice slices_hull_;
101 const string& tag, const TensorSlice& slice,
Dtensor_slice_set.cc33 Status TensorSliceSet::Register(const TensorSlice& slice, const string& tag, in Register()
63 bool TensorSliceSet::Query(const TensorSlice& slice, float* data) const { in Query()
90 TensorSlice intersection; in Query()
120 const TensorSlice& slice, in QueryMeta()
121 std::vector<std::pair<TensorSlice, string>>* results) const { in QueryMeta() argument
147 TensorSlice intersection; in QueryMeta()
173 const string& tag, const TensorSlice& slice, in RegisterTensorSlice()
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()
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
228 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.h61 const TensorSlice& slice, const T* data);
100 const TensorSlice& slice, const T* data) { in Add()
/external/tensorflow/tensorflow/core/kernels/
Dsave_op_test.cc146 TensorSlice s = TensorSlice::ParseOrDie("-"); in TEST_F()
165 TensorSlice s = TensorSlice::ParseOrDie("-"); in TEST_F()
184 TensorSlice s = TensorSlice::ParseOrDie("-:-"); in TEST_F()
203 TensorSlice s = TensorSlice::ParseOrDie("-:-"); in TEST_F()
222 TensorSlice s = TensorSlice::ParseOrDie("-:-"); in TEST_F()
240 TensorSlice s = TensorSlice::ParseOrDie("-:-"); in TEST_F()
258 TensorSlice s = TensorSlice::ParseOrDie("-"); in TEST_F()
276 TensorSlice s = TensorSlice::ParseOrDie("-"); in TEST_F()
294 TensorSlice s = TensorSlice::ParseOrDie("-"); in TEST_F()
312 TensorSlice s = TensorSlice::ParseOrDie("-"); in TEST_F()
[all …]
Dload_and_remap_matrix_op.cc169 std::vector<TensorSlice> tensor_slices; in Compute()
170 TensorSlice slice(tensor_shape.dims()); in Compute()
203 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()
284 TensorSlice parsed_slice; in run()
Dpooling_ops_3d.cc242 TensorSlice src{{0, -1}, {c, 1}, {r, 1}, {p, 1}, {0, -1}}; in launch()
243 TensorSlice dst{{0, -1}, in launch()
417 TensorSlice src{{0, -1}, {c, 1}, {r, 1}, {p, 1}, {0, -1}}; in launch()
418 TensorSlice dst{{0, -1}, in launch()
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle_test.cc328 TensorSlice slice1 = TensorSlice::ParseOrDie("-:0,1"); in TEST()
329 TensorSlice slice2 = TensorSlice::ParseOrDie("-:1,9"); in TEST()
361 std::vector<TensorSlice> slices; in TEST()
374 const TensorSlice distinct_slice = TensorSlice::ParseOrDie("-:0,2"); in TEST()
392 const TensorSlice distinct_slice = TensorSlice::ParseOrDie("-:2,2"); in TEST()
406 TensorSlice::ParseOrDie("-:-"), kExpected)); in TEST()
408 TensorSlice::ParseOrDie("0,5:0,10"), in TEST()
416 const TensorSlice slice = TensorSlice::ParseOrDie("-:-"); in TEST()
425 const TensorSlice slice = TensorSlice::ParseOrDie("0,5:0,10"); in TEST()
434 const TensorSlice slice = TensorSlice::ParseOrDie("0,5:0,10"); in TEST()
[all …]
Dtensor_bundle.h141 const TensorSlice& slice_spec, const Tensor& slice_tensor);
236 Status LookupTensorSlices(StringPiece key, std::vector<TensorSlice>* slices)
243 Status LookupSlice(StringPiece full_tensor_key, const TensorSlice& slice_spec,
282 const TensorSlice& slice_spec,
Dtensor_bundle.cc332 bool IsFullSlice(const TensorSlice& slice_spec, in IsFullSlice()
451 const TensorSlice& slice_spec, in AddSlice()
889 /* a full slice */ TensorSlice(TensorShape(entry.shape()).dims()), val); in Lookup()
907 /* a full slice */ TensorSlice(TensorShape(entry.shape()).dims()), val); in ReadCurrent()
912 std::vector<TensorSlice>* slices) { in LookupTensorSlices()
924 const TensorSlice& slice_spec, Tensor* val) { in LookupSlice()
933 const TensorSlice& slice_spec, Tensor* val) { in GetSliceValue()
939 std::vector<std::pair<TensorSlice, string>> details; in GetSliceValue()
953 /* full slice */ TensorSlice(full_shape.dims()), &tensor_slices_)); in GetSliceValue()
958 /* tag */ "", TensorSlice(slice), &tensor_slices_)); in GetSliceValue()
[all …]
/external/tensorflow/tensorflow/core/ops/
Dio_ops.cc116 TensorSlice parsed_slice; in __anone43acdb40302()
230 TensorSlice parsed_slice; in __anone43acdb40802()
/external/tensorflow/tensorflow/c/
Dcheckpoint_reader.cc129 TensorSlice(slice_proto))) in BuildV2VarMaps()

12