Home
last modified time | relevance | path

Searched refs:ConvertToTensor (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dpywrap_dtensor_device.cc65 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 Dconvert_tensor.h60 Status ConvertToTensor(mlir::ElementsAttr attr, Tensor* output_tensor);
H A Deval_util.cc108 RETURN_FAILURE_IF_ERROR(ConvertToTensor(operand, &tensor)); in EvaluateOperation()
H A Dconvert_tensor_test.cc124 TF_ASSERT_OK(ConvertToTensor(attr, &out)); in VerifyConversion()
H A Dconvert_tensor.cc481 Status ConvertToTensor(const mlir::ElementsAttr attr, Tensor* output_tensor) { in ConvertToTensor() function
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/
H A Dconvert_tensor.h74 tensorflow::Status ConvertToTensor(ElementsAttr attr,
H A Dconvert_tensor.cc480 Status ConvertToTensor(const ElementsAttr attr, Tensor* output_tensor) { in ConvertToTensor() function
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/utils/
H A Deval_utils.cc109 if (!ConvertToTensor(operand, &input_tensor).ok()) return failure(); in EvaluateOperation()
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/utils/
H A Dshape_inference_utils.cc316 auto status = ConvertToTensor(attr, input_tensor); in InferReturnTypeComponentsForTFOp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_with_tf2xla.cc692 auto status = tensorflow::ConvertToTensor(const_attr, &tensor); in GetExprForOperand()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dpywrap_tfe_src.cc3462 bool ConvertToTensor( function
3507 if (!ConvertToTensor(
3870 if (!ConvertToTensor(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dlower_static_tensor_list.cc254 if (!tensorflow::ConvertToTensor(proto_attr, &tensor).ok()) in matchAndRewrite()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_export.cc762 auto status = tensorflow::ConvertToTensor(attr, &tensor); in BuildBuffer()