Lines Matching refs:slice
40 const TensorSlice& slice, SavedSlice* ss);
80 TensorSlice slice = TensorSlice::ParseOrDie("-:0,1");
82 TF_CHECK_OK(writer.Add("test", shape, slice, data));
88 TensorSlice slice = TensorSlice::ParseOrDie("-:3,1");
90 TF_CHECK_OK(writer.Add("test", shape, slice, data));
93 // Another slice from a different float tensor -- it has a different name and
97 TensorSlice slice = TensorSlice::ParseOrDie("-:-");
99 TF_CHECK_OK(writer.Add("AA", shape, slice, data));
102 // A slice with int64 data
105 TensorSlice slice = TensorSlice::ParseOrDie("-:3,1");
107 TF_CHECK_OK(writer.Add("int64", shape, slice, data));
110 // A slice with int16 data
113 TensorSlice slice = TensorSlice::ParseOrDie("-:3,1");
115 TF_CHECK_OK(writer.Add("int16", shape, slice, data));
128 const TensorSlice& slice,
130 string key = EncodeTensorNameSlice(name, slice);
138 TensorSlice slice2(ss->slice());
139 EXPECT_EQ(slice.DebugString(), slice2.DebugString());
176 TensorSlice s0(ssm.slice(0));
177 TensorSlice s1(ssm.slice(1));
191 TensorSlice s0(ssm.slice(0));
204 TensorSlice s0(ssm.slice(0));
217 TensorSlice s0(ssm.slice(0));
224 // Block 1: we expect it to be the full slice of the "AA" tensor
234 // Block 2: we expect it to be the first slice of the "test" tensor
243 // Block 3: we expect it to be the second slice of the "test" tensor
252 // Block 4: we expect it to be the slice of the "int64" tensor
262 // Block 5: we expect it to be the slice of the "int16" tensor
329 // Add a 300MB int8 tensor slice, which will fail because it expands to 3GB.
332 TensorSlice slice = TensorSlice::ParseOrDie("-:-");
334 Status s = writer.Add("test1", shape, slice, data.data());
337 .contains("Tensor slice is too large to serialize"));
340 // Add a large string tensor slice, which will fail.
343 TensorSlice slice = TensorSlice::ParseOrDie("-:-");
345 Status s = writer.Add("test2", shape, slice, data.data());
348 .contains("Tensor slice is too large to serialize"));
