OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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