HomeSort by relevance Sort by last modified time
    Searched refs:tensor_dtype (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/contrib/framework/python/ops/
accumulate_n_v2.py 31 def accumulate_n_v2(inputs, shape=None, tensor_dtype=None, name=None):
34 Optionally, pass `shape` and `tensor_dtype` for shape and type checking,
52 tf.accumulate_n_v2([a, b, a], shape=[2, 2], tensor_dtype=tf.int32)
60 tensor_dtype: The type of `inputs`.
87 # tensor_dtype is for safety only; operator's output type computed in C++
88 if tensor_dtype is not None and tensor_dtype != inputs[0].dtype:
89 raise TypeError("tensor_dtype is {}, but input is of type {}"
90 .format(tensor_dtype, inputs[0].dtype))
accumulate_n_v2_test.py 111 tf_val = av2.accumulate_n_v2([a, b], tensor_dtype=np.int32)
118 tf_val = av2.accumulate_n_v2([a], tensor_dtype=np.int32)
  /external/tensorflow/tensorflow/python/framework/
tensor_util.py 556 tensor_dtype = dtypes.as_dtype(tensor.dtype)
557 dtype = tensor_dtype.as_numpy_dtype
562 elif tensor_dtype == dtypes.float16:
573 elif tensor_dtype == dtypes.float32:
580 elif tensor_dtype == dtypes.float64:
587 elif tensor_dtype in [
597 elif tensor_dtype == dtypes.int64:
604 elif tensor_dtype == dtypes.string:
612 elif tensor_dtype == dtypes.complex64:
623 elif tensor_dtype == dtypes.complex128
    [all...]
  /external/tensorflow/tensorflow/python/ops/
math_ops.py     [all...]
  /external/tensorflow/tensorflow/core/debug/
debug_io_utils.cc 61 const DataType& tensor_dtype,
102 value->mutable_tensor()->set_dtype(tensor_dtype);
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/
cwise_ops_test.py     [all...]

Completed in 3608 milliseconds