/external/tensorflow/tensorflow/core/framework/ |
D | tensor_slice_test.cc | 30 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 …]
|
D | tensor_slice.cc | 26 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 …]
|
D | tensor_slice.h | 34 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/ |
D | tensor_slice_set_test.cc | 53 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 …]
|
D | tensor_slice_reader_test.cc | 72 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 …]
|
D | tensor_slice_util_test.cc | 46 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()
|
D | tensor_slice_writer_test.cc | 41 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 …]
|
D | tensor_slice_set.h | 55 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,
|
D | tensor_slice_set.cc | 33 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()
|
D | tensor_slice_reader.h | 91 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()
|
D | saved_tensor_slice_util_test.cc | 33 TensorSlice s = TensorSlice::ParseOrDie("-:-:1,3:4,5"); in TEST()
|
D | saved_tensor_slice_util.cc | 28 string EncodeTensorNameSlice(const string& name, const TensorSlice& slice) { in EncodeTensorNameSlice() 46 tensorflow::TensorSlice* slice) { in DecodeTensorNameSlice() 91 TensorSlice* slice, TensorShape* shape_slice) { in ParseShapeAndSlice()
|
D | tensor_slice_util.h | 138 const TensorSlice& slice_s, 139 const TensorSlice& slice_d, 146 TensorSlice inter; 168 TensorSlice rel_s, rel_d;
|
D | tensor_slice_reader.cc | 173 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()
|
D | saved_tensor_slice_util.h | 47 const tensorflow::TensorSlice& slice); 51 tensorflow::TensorSlice* slice); 57 TensorSlice* slice, TensorShape* shape_slice);
|
D | tensor_slice_writer.h | 61 const TensorSlice& slice, const T* data); 100 const TensorSlice& slice, const T* data) { in Add()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | save_op_test.cc | 146 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 …]
|
D | load_and_remap_matrix_op.cc | 169 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()
|
D | save_restore_tensor.cc | 104 TensorSlice slice(input.dims()); in SaveTensors() 193 TensorSlice slice_to_load(saved_shape.dims()); in RestoreTensor() 284 TensorSlice parsed_slice; in run()
|
D | pooling_ops_3d.cc | 242 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/ |
D | tensor_bundle_test.cc | 328 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 …]
|
D | tensor_bundle.h | 141 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,
|
D | tensor_bundle.cc | 332 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/ |
D | io_ops.cc | 116 TensorSlice parsed_slice; in __anone43acdb40302() 230 TensorSlice parsed_slice; in __anone43acdb40802()
|
/external/tensorflow/tensorflow/c/ |
D | checkpoint_reader.cc | 129 TensorSlice(slice_proto))) in BuildV2VarMaps()
|