Home
last modified time | relevance | path

Searched refs:MaxBytesPerElement (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dtensor_slice_writer_test.cc291 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_BOOL), in TEST()
293 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_BOOL), in TEST()
295 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_FLOAT), in TEST()
297 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_DOUBLE), in TEST()
299 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_COMPLEX64), in TEST()
301 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_COMPLEX128), in TEST()
303 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_INT32), in TEST()
305 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_INT64), in TEST()
307 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_UINT16), in TEST()
309 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_UINT8), in TEST()
[all …]
Dtensor_slice_writer.cc133 size_t TensorSliceWriter::MaxBytesPerElement(DataType dt) { in MaxBytesPerElement() function in tensorflow::checkpoint::TensorSliceWriter
182 (num_elements * MaxBytesPerElement(DT_INT32)); in SaveData()
Dtensor_slice_writer.h68 static size_t MaxBytesPerElement(DataType dt);
167 (MaxBytesPerElement(DataTypeToEnum<T>::value) * num_elements); in SaveData()