Searched refs:ConvertToTensor (Results 1 – 13 of 13) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/python/ |
H A D | pywrap_dtensor_device.cc | 65 void ConvertToTensor(TFE_Context* ctx, PyObject* input, in ConvertToTensor() function 223 ConvertToTensor(ctx, elem, &py_eager_tensor_handles[i], status.get()); in PYBIND11_MODULE()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
H A D | convert_tensor.h | 60 Status ConvertToTensor(mlir::ElementsAttr attr, Tensor* output_tensor);
|
H A D | eval_util.cc | 108 RETURN_FAILURE_IF_ERROR(ConvertToTensor(operand, &tensor)); in EvaluateOperation()
|
H A D | convert_tensor_test.cc | 124 TF_ASSERT_OK(ConvertToTensor(attr, &out)); in VerifyConversion()
|
H A D | convert_tensor.cc | 481 Status ConvertToTensor(const mlir::ElementsAttr attr, Tensor* output_tensor) { in ConvertToTensor() function
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/ |
H A D | convert_tensor.h | 74 tensorflow::Status ConvertToTensor(ElementsAttr attr,
|
H A D | convert_tensor.cc | 480 Status ConvertToTensor(const ElementsAttr attr, Tensor* output_tensor) { in ConvertToTensor() function
|
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/utils/ |
H A D | eval_utils.cc | 109 if (!ConvertToTensor(operand, &input_tensor).ok()) return failure(); in EvaluateOperation()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/utils/ |
H A D | shape_inference_utils.cc | 316 auto status = ConvertToTensor(attr, input_tensor); in InferReturnTypeComponentsForTFOp()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
H A D | legalize_tf_with_tf2xla.cc | 692 auto status = tensorflow::ConvertToTensor(const_attr, &tensor); in GetExprForOperand()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
H A D | pywrap_tfe_src.cc | 3462 bool ConvertToTensor( function 3507 if (!ConvertToTensor( 3870 if (!ConvertToTensor(
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
H A D | lower_static_tensor_list.cc | 254 if (!tensorflow::ConvertToTensor(proto_attr, &tensor).ok()) in matchAndRewrite()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
H A D | flatbuffer_export.cc | 762 auto status = tensorflow::ConvertToTensor(attr, &tensor); in BuildBuffer()
|