Home
last modified time | relevance | path

Searched refs:TF_STRING (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
Dc_api_session_coordination_test.cc96 TF_Operation* feed_key = Placeholder(graph, status, "key", TF_STRING, {}); in TEST_P()
98 TF_Operation* feed_val = Placeholder(graph, status, "val", TF_STRING, {}); in TEST_P()
130 input_values[0] = TF_NewTensor(TF_STRING, nullptr, 0, &tstr_key, in TEST_P()
132 input_values[1] = TF_NewTensor(TF_STRING, nullptr, 0, &tstr_val, in TEST_P()
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor.cc169 *out_tf_datatype = TF_STRING; in PyArray_TYPE_to_TF_DataType()
285 DCHECK_EQ(TF_STRING, TF_TensorType(src)); in CopyTF_TensorStringsToPyArray()
407 if (dtype == TF_STRING || dtype == TF_RESOURCE) { in TF_TensorToMaybeAliasedPyArray()
460 if (TF_TensorType(tensor.get()) == TF_STRING) { in TF_TensorToPyArray()
517 } else if (dtype != TF_STRING) { in NdarrayToTensor()
Dndarray_tensor_bridge.cc161 case TF_STRING: in TF_DataType_to_PyArray_TYPE()
/external/tensorflow/tensorflow/dtensor/cc/
Dsmall_constant_optimization.cc41 TF_STRING};
118 case TF_STRING: in ExtractSmallTensorValue()
Ddtensor_device.cc1926 if (!(num_dims == 0 || dtype == TF_STRING || small_int_tensor)) { in Execute()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil.cc100 return TF_STRING; in GetTensorFlowDataType()
134 case TF_STRING: in GetTensorFlowLiteType()
Dutil_test.cc128 EXPECT_EQ(TF_STRING, GetTensorFlowDataType(kTfLiteString)); in TEST()
145 EXPECT_EQ(kTfLiteString, GetTensorFlowLiteType(TF_STRING)); in TEST()
/external/tensorflow/tensorflow/c/
Dtf_datatype.h54 TF_STRING = 7, enumerator
Dtf_tensor.cc105 if (dtype != TF_STRING && dtype != TF_RESOURCE && in TF_NewTensor()
Dc_api_test.cc305 EXPECT_EQ(TF_STRING, static_cast<TF_DataType>(tensorflow::DT_STRING)); in TEST()
321 EXPECT_EQ(TF_DataTypeSize(TF_STRING), in TEST()
2361 unique_tensor_ptr t_in(TF_NewTensor(TF_STRING, nullptr, 0, &tstr[0], in TEST_F()
2378 EXPECT_EQ(TF_STRING, TF_TensorType(t_out)); in TEST_F()
2652 TF_Tensor* t = TF_AllocateTensor(TF_STRING, dims, num_dims, in TEST()
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc332 TF_Tensor* t = TF_AllocateTensor(TF_STRING, nullptr, 0, sizeof(TF_TString)); in Java_org_tensorflow_Tensor_allocateScalarBytes()
407 TF_Tensor* t = TF_AllocateTensor(TF_STRING, dims, num_dims, in Java_org_tensorflow_Tensor_allocateNonScalarBytes()
523 if (TF_TensorType(t) != TF_STRING) { in Java_org_tensorflow_Tensor_scalarBytes()
549 if (dtype == TF_STRING) { in Java_org_tensorflow_Tensor_readNDArray()
/external/tensorflow/tensorflow/go/
Dtensor.go53 String DataType = C.TF_STRING
403 {reflect.TypeOf(""), C.TF_STRING},
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1830 #define TF_STRING(s) TF_XSTRING(s) in GetSupportedCudaComputeCapabilities() macro
1831 string extra_cuda_caps = TF_STRING(TF_EXTRA_CUDA_CAPABILITIES); in GetSupportedCudaComputeCapabilities()
1832 #undef TF_STRING in GetSupportedCudaComputeCapabilities()
/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc49 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_STRING, nullptr, 0, status); in TestScalarTensorHandle()
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc1218 .value("TF_STRING", TF_STRING) in PYBIND11_MODULE()