HomeSort by relevance Sort by last modified time
    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