Home
last modified time | relevance | path

Searched refs:TF_FLOAT (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc36 TF_FLOAT, &dims[0], sizeof(dims) / sizeof(int64_t), sizeof(data)); in TestMatrixTensorHandle()
192 EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(h)); in TEST()
870 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in CreateVariable()
885 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in CreateVariable()
890 TF_AllocateTensor(TF_FLOAT, nullptr, 0, sizeof(value)), TF_DeleteTensor); in CreateVariable()
924 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in TEST()
934 EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(value_handle)); in TEST()
964 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in BM_ReadVariable()
976 CHECK_EQ(TF_FLOAT, TFE_TensorHandleDataType(h)); in BM_ReadVariable()
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc50 case TF_FLOAT: in elemByteSize()
92 CASE(TF_FLOAT, jfloat, "floatValue", "()F", Float); in writeScalar()
136 CASE(TF_FLOAT, jfloat, Float); in write1DArray()
171 CASE(TF_FLOAT, jfloat, Float); in read1DArray()
548 DEFINE_GET_SCALAR_METHOD(jfloat, TF_FLOAT, Float);
/external/tensorflow/tensorflow/c/
Dc_api_test.cc130 TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes, in TEST()
133 EXPECT_EQ(TF_FLOAT, TF_TensorType(t)); in TEST()
150 TF_NewTensor(TF_FLOAT, nullptr, 0, nullptr, 0, &NoOpDeallocator, nullptr); in TEST()
157 TF_Tensor* t = TF_AllocateTensor(TF_FLOAT, dims, 2, num_bytes); in TEST()
158 EXPECT_EQ(TF_FLOAT, TF_TensorType(t)); in TEST()
173 TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes, in TEST()
272 EXPECT_EQ(TF_FLOAT, static_cast<TF_DataType>(tensorflow::DT_FLOAT)); in TEST()
1389 EXPECT_EQ(TF_FLOAT, TF_TensorType(out)); in TEST()
1631 TF_Tensor* t = TF_AllocateTensor(TF_FLOAT, dims, 2, sizeof(float) * 4); in FloatTensor2x2()
1642 TF_SetAttrType(desc, "dtype", TF_FLOAT); in FloatConst2x2()
[all …]
Dc_api.h99 TF_FLOAT = 1, enumerator
Dc_api_function_test.cc1514 RandomUniform(shape, TF_FLOAT, func_graph.get(), s.get()); in DefineStatefulFunction()
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor.cc78 *out_tf_datatype = TF_FLOAT; in PyArray_TYPE_to_TF_DataType()
383 TF_DataType dtype = TF_FLOAT; in PyArrayToTF_Tensor()
Dndarray_tensor_bridge.cc115 case TF_FLOAT: in TF_DataType_to_PyArray_TYPE()
/external/tensorflow/tensorflow/go/
Dtensor.go39 Float DataType = C.TF_FLOAT
222 {reflect.TypeOf(float32(0)), C.TF_FLOAT},