HomeSort by relevance Sort by last modified time
    Searched refs:LookupTensorShape (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
save_v2_op_test.cc 133 TF_EXPECT_OK(reader.LookupTensorShape("tensor_bool", &shape));
149 TF_EXPECT_OK(reader.LookupTensorShape("tensor_int", &shape));
165 TF_EXPECT_OK(reader.LookupTensorShape("tensor_float", &shape));
181 TF_EXPECT_OK(reader.LookupTensorShape("tensor_double", &shape));
197 TF_EXPECT_OK(reader.LookupTensorShape("tensor_qint8", &shape));
213 TF_EXPECT_OK(reader.LookupTensorShape("tensor_qint32", &shape));
230 TF_EXPECT_OK(reader.LookupTensorShape("tensor_uint8", &shape));
246 TF_EXPECT_OK(reader.LookupTensorShape("tensor_int8", &shape));
262 TF_EXPECT_OK(reader.LookupTensorShape("tensor_int16", &shape));
278 TF_EXPECT_OK(reader.LookupTensorShape("tensor_int64", &shape))
    [all...]
save_restore_tensor.cc 259 reader.LookupTensorShape(tensor_name, &restored_full_shape));
  /external/tensorflow/tensorflow/core/util/tensor_bundle/
tensor_bundle.h 204 Status LookupTensorShape(StringPiece key,
tensor_bundle.cc     [all...]

Completed in 214 milliseconds