| /external/tensorflow/tensorflow/python/framework/ |
| dtypes_test.py | 51 dt = dtypes.as_dtype(datatype_enum)
58 dtype = dtypes.as_dtype(datatype_enum)
64 dtypes.as_dtype(datatype_enum).base_dtype,
65 dtypes.as_dtype(numpy_dtype))
71 dtypes.as_dtype(types_pb2.DT_INVALID)
74 self.assertIs(dtypes.float32, dtypes.as_dtype(np.float32))
75 self.assertIs(dtypes.float64, dtypes.as_dtype(np.float64))
76 self.assertIs(dtypes.int32, dtypes.as_dtype(np.int32))
77 self.assertIs(dtypes.int64, dtypes.as_dtype(np.int64))
78 self.assertIs(dtypes.uint8, dtypes.as_dtype(np.uint8)
[all...] |
| op_def_library.py | 59 (param_name, dtypes.as_dtype(dtype).name,
60 ", ".join(dtypes.as_dtype(x).name for x in allowed_list)))
174 v = dtypes.as_dtype(v).base_dtype
386 default_type_attr_map[key] = dtypes.as_dtype(
449 dtype = dtypes.as_dtype(dtype).base_dtype
518 (dtypes.as_dtype(dtype).name, input_arg.name, op_type_name,
533 (prefix, dtypes.as_dtype(input_arg.type).name))
545 (prefix, dtypes.as_dtype(attrs[input_arg.type_attr]).name,
621 ", ".join(dtypes.as_dtype(x).name for x in attr_value),
622 ", ".join(dtypes.as_dtype(x).nam
[all...] |
| tensor_spec.py | 52 self._dtype = dtypes.as_dtype(dtype)
168 dtype = dtypes.as_dtype(spec.dtype)
|
| sparse_tensor.py | 248 dtype = dtypes.as_dtype(dtype)
|
| dtypes.py | 62 The `tf.as_dtype()` function converts numpy types and string type
71 `tf.as_dtype()` function.
265 other = as_dtype(other)
274 dtype = as_dtype(other).as_datatype_enum
650 @tf_export("as_dtype")
651 def as_dtype(type_value): function
|
| /external/tensorflow/tensorflow/contrib/remote_fused_graph/pylib/python/ops/ |
| remote_fused_graph_ops.py | 53 type_shape_proto.dtype = int(dtypes.as_dtype(type_shape[0]))
59 type_shape_proto.dtype = int(dtypes.as_dtype(type_shape[0]))
|
| /external/tensorflow/tensorflow/python/estimator/export/ |
| export_output.py | 95 and dtypes.as_dtype(classes.dtype) == dtypes.string)):
117 if dtypes.as_dtype(examples.dtype) != dtypes.string:
151 if dtypes.as_dtype(examples.dtype) != dtypes.string:
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| reverse_sequence_op_test.py | 39 p = array_ops.placeholder(dtypes.as_dtype(x.dtype))
40 lengths = array_ops.placeholder(dtypes.as_dtype(seq_lengths.dtype))
|
| dynamic_stitch_test.py | 35 array_ops.placeholder(dtypes.as_dtype(arg.dtype)) for arg in indices
38 array_ops.placeholder(dtypes.as_dtype(arg.dtype)) for arg in data
|
| xla_test.py | 57 dtypes.as_dtype(types_pb2.DataType.Value(name))
112 dtypes.as_dtype(types_pb2.DataType.Value(name)).as_numpy_dtype
119 tf_types = set([dtypes.as_dtype(t)
136 tf_types = set([dtypes.as_dtype(t)
|
| ternary_ops_test.py | 35 pa = array_ops.placeholder(dtypes.as_dtype(a.dtype), a.shape, name="a")
36 pb = array_ops.placeholder(dtypes.as_dtype(b.dtype), b.shape, name="b")
37 pc = array_ops.placeholder(dtypes.as_dtype(c.dtype), c.shape, name="c")
|
| argminmax_test.py | 43 dtypes.as_dtype(inp.dtype), inp.shape, name="a")
|
| fft_test.py | 76 dtypes.as_dtype(data.dtype), shape=data.shape)
98 dtypes.as_dtype(data.dtype), shape=data.shape)
|
| cholesky_op_test.py | 54 dtypes.as_dtype(x.dtype), shape=x.shape)
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| utils_impl.py | 44 dtype=dtypes.as_dtype(tensor.dtype).as_datatype_enum,
|
| /external/tensorflow/tensorflow/python/ops/ |
| init_ops.py | 94 self.dtype = dtypes.as_dtype(dtype)
110 self.dtype = dtypes.as_dtype(dtype)
208 self.dtype = dtypes.as_dtype(dtype)
247 self.dtype = dtypes.as_dtype(dtype)
284 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
326 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
378 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
449 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
510 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
558 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype)
[all...] |
| gradients_impl.py | 248 (dtypes.as_dtype(grad_y.dtype).name, y,
249 dtypes.as_dtype(y.dtype).name))
254 (dtypes.as_dtype(grad_y.dtype).name, y,
255 dtypes.as_dtype(y.dtype).name))
259 (y, dtypes.as_dtype(y.dtype).name))
284 dtype = dtypes.as_dtype(tensor.dtype)
[all...] |
| lookup_ops.py | 109 self._key_dtype = dtypes.as_dtype(key_dtype)
110 self._value_dtype = dtypes.as_dtype(value_dtype)
298 self._key_dtype = dtypes.as_dtype(key_dtype)
299 self._value_dtype = dtypes.as_dtype(value_dtype)
467 key_dtype = dtypes.as_dtype(key_dtype)
468 value_dtype = dtypes.as_dtype(value_dtype)
[all...] |
| gradient_checker.py | 219 t = dtypes.as_dtype(x.dtype)
223 t2 = dtypes.as_dtype(y.dtype)
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| multitask_optimizer_wrapper.py | 133 lambda: array_ops.zeros([], dtype=dtypes.as_dtype(grad.dtype)),
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| iterator_ops.py | 156 output_types = nest.map_structure(dtypes.as_dtype, output_types)
227 output_types = nest.map_structure(dtypes.as_dtype, output_types)
|
| /external/tensorflow/tensorflow/python/estimator/inputs/queues/ |
| feeding_functions.py | 387 types = [dtypes.int64, dtypes.as_dtype(data.dtype)]
392 ] + [dtypes.as_dtype(col.dtype) for col in data.values()]
399 types = [dtypes.as_dtype(col.dtype) for col in x_first_values]
404 dtypes.as_dtype(dt) for dt in [data.index.dtype] + list(data.dtypes)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| matrix_solve_op_test.py | 59 a_ph = array_ops.placeholder(dtypes.as_dtype(np_type))
60 b_ph = array_ops.placeholder(dtypes.as_dtype(np_type))
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| linear_operator_test_util.py | 64 dtype = dtypes.as_dtype(x.dtype)
442 dtype = dtypes.as_dtype(dtype)
510 dtype = dtypes.as_dtype(dtype)
548 dtype = dtypes.as_dtype(dtype)
591 dtype = dtypes.as_dtype(dtype)
646 dtype = dtypes.as_dtype(dtype)
|
| /external/tensorflow/tensorflow/tools/quantization/ |
| quantize_graph.py | 568 dtype = dtypes.as_dtype(node.attr["dtype"].type)
592 dtypes.as_dtype(current_node.attr["T"].type) == dtypes.float32)
595 dtypes.as_dtype(current_node.attr["T"].type) == dtypes.float32)
618 dtypes.as_dtype(current_node.attr["T"].type) == dtypes.float32):
623 dtypes.as_dtype(current_node.attr["T"].type) == dtypes.float32):
[all...] |
