Home
last modified time | relevance | path

Searched refs:tensor_dtype (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dpad_test.cc319 template <typename integer_type, TensorType tensor_dtype>
324 PadOpConstModel m({tensor_dtype, {1, 2, 2, 1}, 1.0, 2.0}, {4, 2}, in ZeroNotInQuantizationRange()
325 {0, 0, 1, 1, 1, 1, 0, 0}, {tensor_dtype, {}, 1.0, 2.0}), in ZeroNotInQuantizationRange()
337 template <typename integer_type, TensorType tensor_dtype>
341 PadOpConstModel m({tensor_dtype, {1, 2, 2, 1}, -1.0, 1.0}, {4, 2}, in SimpleConstTest()
342 {0, 0, 1, 1, 1, 1, 0, 0}, {tensor_dtype, {}, -1.0, 1.0}); in SimpleConstTest()
359 template <typename integer_type, TensorType tensor_dtype>
361 PadOpDynamicModel m({tensor_dtype, {1, 2, 2, 1}, -1.0, 1.0}, {4, 2}, in SimpleDynamicTest()
362 {tensor_dtype, {}, -1.0, 1.0}); in SimpleDynamicTest()
380 template <typename integer_type, TensorType tensor_dtype>
[all …]
/external/tensorflow/tensorflow/python/framework/
Dtensor_util.py598 tensor_dtype = dtypes.as_dtype(tensor.dtype)
599 dtype = tensor_dtype.as_numpy_dtype
605 if tensor_dtype == dtypes.string:
614 if tensor_dtype == dtypes.float16 or tensor_dtype == dtypes.bfloat16:
618 values.dtype = tensor_dtype.as_numpy_dtype
619 elif tensor_dtype == dtypes.float32:
621 elif tensor_dtype == dtypes.float64:
623 elif tensor_dtype in [
628 elif tensor_dtype == dtypes.int64:
630 elif tensor_dtype == dtypes.complex64:
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Daccumulate_n_test.py122 math_ops.accumulate_n([a, b], tensor_dtype=np.int32)
129 math_ops.accumulate_n([a], tensor_dtype=np.int32)
Dcwise_ops_test.py1062 math_ops.accumulate_n([a, b], tensor_dtype=np.int32)
1069 math_ops.accumulate_n([a], tensor_dtype=np.int32)
/external/tensorflow/tensorflow/python/ops/
Dmath_ops.py2803 def accumulate_n(inputs, shape=None, tensor_dtype=None, name=None): argument
2863 if tensor_dtype is not None and tensor_dtype != inputs[0].dtype:
2865 tensor_dtype, inputs[0].dtype))
/external/tensorflow/tensorflow/core/debug/
Ddebug_io_utils.cc64 const DataType& tensor_dtype, in PrepareChunkEventProto() argument
105 value->mutable_tensor()->set_dtype(tensor_dtype); in PrepareChunkEventProto()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.pbtxt9 …argspec: "args=[\'inputs\', \'shape\', \'tensor_dtype\', \'name\'], varargs=None, keywords=None, d…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.pbtxt9 …argspec: "args=[\'inputs\', \'shape\', \'tensor_dtype\', \'name\'], varargs=None, keywords=None, d…
Dtensorflow.pbtxt697 …argspec: "args=[\'inputs\', \'shape\', \'tensor_dtype\', \'name\'], varargs=None, keywords=None, d…