Searched refs:handle_dtype (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_util.cc | 302 DataType handle_dtype; in GetLookupTable() local
303 TF_RETURN_IF_ERROR(ctx->input_dtype(input_name, &handle_dtype)); in GetLookupTable()
304 if (handle_dtype == DT_RESOURCE) { in GetLookupTable()
314 DataType handle_dtype; in GetInitializableLookupTable() local
315 TF_RETURN_IF_ERROR(ctx->input_dtype(input_name, &handle_dtype)); in GetInitializableLookupTable()
316 if (handle_dtype == DT_RESOURCE) { in GetInitializableLookupTable()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | cpp_shape_inference.proto | 28 reserved 3; // was handle_dtype
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 123 Type GetTensorTypeForTensorList(Type element_type, TF::VariantType handle_dtype, in GetTensorTypeForTensorList() argument
128 if (handle_dtype.getSubtypes().empty()) { in GetTensorTypeForTensorList()
131 return PrependLeadingDimIfRanked(-1, handle_dtype.getSubtypes()[0], rewriter); in GetTensorTypeForTensorList()
535 auto handle_dtype = getElementTypeOrSelf(op.output_handle().getType()) in matchAndRewrite() local
538 GetTensorTypeForTensorList(elem_type, handle_dtype, &rewriter); in matchAndRewrite()
579 auto handle_dtype = getElementTypeOrSelf(op.output_handle().getType()) in matchAndRewrite() local
582 GetTensorTypeForTensorList(elem_type, handle_dtype, &rewriter); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 283 TF_DataType handle_dtype = TFE_TensorHandleDataType(handle.get()); in ConvertToEagerTensorUncached() local
285 dtype != static_cast<DataType>(handle_dtype)) { in ConvertToEagerTensorUncached()
286 if (tensorflow::IsCompatible(dtype, static_cast<DataType>(handle_dtype))) { in ConvertToEagerTensorUncached()
288 tensorflow::EagerCast(ctx, handle.get(), handle_dtype, in ConvertToEagerTensorUncached()
294 static_cast<DataType>(handle_dtype)), in ConvertToEagerTensorUncached()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 50 if (handle_dtype().getSubtypes().size() != 1) {
63 TensorType element_type() { return handle_dtype().getSubtypes()[0]; }
67 VariantType handle_dtype() {
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 3133 DataType handle_dtype = arg_shapes[orig_arg_num].handle_type; in BuildExecuteNodes() local
3135 handle_dtype) == kTpuAllTypes.end()) { in BuildExecuteNodes()
3138 DataTypeString(handle_dtype), ", caused by output ", in BuildExecuteNodes()
|