Home
last modified time | relevance | path

Searched defs:ndarray (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/experimental/quantization_debugger/
Ddebugger.py53 Callable[[np.ndarray],
56 str, Callable[[Sequence[np.ndarray], Sequence[np.ndarray]],
90 Iterable[Sequence[np.ndarray]]]] = None,
230 tensor_data: Sequence[np.ndarray],
/external/tensorflow/tensorflow/python/tpu/
Ddevice_assignment.py70 def __init__(self, topology: Topology, core_assignment: np.ndarray):
180 computation_shape: Optional[np.ndarray] = None,
181 computation_stride: Optional[np.ndarray] = None,
339 computation_shape: Optional[np.ndarray] = None,
340 computation_stride: Optional[np.ndarray] = None,
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_arrays.py54 ndarray = ops.Tensor variable
/external/tensorflow/tensorflow/compiler/xla/python_api/
Dxla_literal.py64 def _ConvertNumpyArrayToLiteral(ndarray): argument
Dxla_shape.py103 def _CreateShapeFromNumpy(ndarray): # pylint: disable=invalid-name argument
/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
Dmodel_handler.py176 output_tensors: Sequence[np.ndarray],
299 inputs: Optional[Mapping[str, np.ndarray]] = None,
424 calibration_inputs: Optional[Mapping[str, np.ndarray]] = None,
471 calibration_inputs: Optional[Mapping[str, np.ndarray]] = None,
492 inputs: Optional[Mapping[str, np.ndarray]] = None,
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc190 jobjectArray ndarray = static_cast<jobjectArray>(src); in writeNDArray() local
209 jobjectArray ndarray = static_cast<jobjectArray>(dst); in readNDArray() local
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dtensor_jni.cc220 jobjectArray ndarray = static_cast<jobjectArray>(dst); in ReadMultiDimensionalArray() local
275 jobjectArray ndarray = static_cast<jobjectArray>(src); in WriteMultiDimensionalArray() local
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor.cc472 Status NdarrayToTensor(TFE_Context* ctx, PyObject* ndarray, in NdarrayToTensor()
/external/python/cpython3/Lib/test/
Dtest_buffer.py29 ndarray = None variable
/external/tensorflow/tensorflow/compiler/xla/python/
Djax_jit.cc494 const auto& ndarray = numpy.attr("ndarray"); in ArgSignatureOfValue() local