OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:TF_DataTypeSize
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/c/kernels/
bitcast_op.cc
62
kernel->in_size =
TF_DataTypeSize
(kernel->input_data_type);
63
kernel->out_size =
TF_DataTypeSize
(kernel->output_data_type);
121
TF_DataTypeSize
(k->output_data_type));
/external/tensorflow/tensorflow/c/
c_api_test.cc
313
EXPECT_EQ(
TF_DataTypeSize
(TF_DOUBLE),
315
EXPECT_EQ(
TF_DataTypeSize
(TF_STRING),
318
EXPECT_EQ(
TF_DataTypeSize
(static_cast<TF_DataType>(0)), 0);
[
all
...]
c_api.cc
112
size_t
TF_DataTypeSize
(TF_DataType dt) {
235
size_t elem_size =
TF_DataTypeSize
(dtype);
300
size_t in_size =
TF_DataTypeSize
(TF_TensorType(from));
306
size_t out_size =
TF_DataTypeSize
(type);
[
all
...]
c_api.h
126
//
TF_DataTypeSize
returns the sizeof() for the underlying type corresponding
129
TF_CAPI_EXPORT extern size_t
TF_DataTypeSize
(TF_DataType dt);
285
// Specifically, given from_type_size =
TF_DataTypeSize
(TF_TensorType(from)):
287
// ShapeElementCount(dims, num_dims) *
TF_DataTypeSize
(type)
297
// *
TF_DataTypeSize
(TF_TensorType(from)) != 0
298
// *
TF_DataTypeSize
(type) != 0
[
all
...]
Completed in 56 milliseconds