Home
last modified time | relevance | path

Searched defs:py_dtype (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/
Dconversion_utils.cc116 auto py_dtype = [](pybind11::dtype dtype) -> char { in ConvertPyArrayMemrefDesc() local
Dtf_cpurt_executor.cc167 auto py_dtype = [](pybind11::dtype dtype) -> char { in ConvertPyArrayMemrefDesc() local
/external/tensorflow/tensorflow/python/framework/
Dpython_tensor_converter.cc43 Safe_PyObjectPtr py_dtype(PyObject_GetAttr(tensor, dtype_attr)); in DataTypeForTensor() local
Dpython_api_parameter_converter.cc115 Safe_PyObjectPtr py_dtype = GetAttr_DType(value); in InferDType() local
254 PyObject* py_dtype = params[input.type_attr->index]; in ConvertInputsWithTypeAttr() local
Dpython_op_gen.cc482 const string py_dtype = in GenerateTypeVars() local
1023 const string py_dtype = in AddEagerInferredAttrs() local
1321 const string py_dtype = in GetArgAnnotation() local
/external/pytorch/torch/csrc/dynamo/
Dguards.cpp774 PyObject* py_dtype = PyTuple_GET_ITEM(args, 2); in _parse_empty_strided_args() local