Searched refs:TFE_TensorHandleDataType (Results 1 – 5 of 5) sorted by relevance
270 TF_DataType handle_dtype = TFE_TensorHandleDataType(handle.get()); in EagerTensor_init()285 handle_dtype = TFE_TensorHandleDataType(handle.get()); in EagerTensor_init()345 return PyIntFromDataType(TFE_TensorHandleDataType(self->handle)); in EagerTensor_datatype_enum()
1376 auto dtype = TFE_TensorHandleDataType(input_handle);1681 attr_value[j] = TFE_TensorHandleDataType(input_handle);
58 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(a)); in MatMulOp()90 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(input)); in MinOp()192 EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(h)); in TEST()934 EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(value_handle)); in TEST()976 CHECK_EQ(TF_FLOAT, TFE_TensorHandleDataType(h)); in BM_ReadVariable()
121 TF_CAPI_EXPORT extern TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h);
153 TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h) { in TFE_TensorHandleDataType() function