| /external/tensorflow/tensorflow/contrib/quantization/ |
| __init__.py | 27 from tensorflow.python.ops import gen_array_ops as quantized_gen_array_ops
28 from tensorflow.python.ops.gen_array_ops import dequantize
29 from tensorflow.python.ops.gen_array_ops import quantize_v2
30 from tensorflow.python.ops.gen_array_ops import quantized_concat
|
| /external/tensorflow/tensorflow/contrib/quantization/python/ |
| array_ops.py | 22 from tensorflow.python.ops import gen_array_ops as quantized_gen_array_ops
23 from tensorflow.python.ops.gen_array_ops import dequantize
24 from tensorflow.python.ops.gen_array_ops import quantize_v2
25 from tensorflow.python.ops.gen_array_ops import quantized_concat
|
| /external/tensorflow/tensorflow/python/layers/ |
| maxout.py | 26 from tensorflow.python.ops import gen_array_ops
107 shape[i] = gen_array_ops.shape(inputs)[i]
109 gen_array_ops.reshape(inputs, shape), -1, keepdims=False)
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| drop_stale_gradient_optimizer.py | 25 from tensorflow.python.ops import gen_array_ops
69 loss = gen_array_ops.identity(loss)
91 return gen_array_ops.identity(0.0)
94 return gen_array_ops.identity(1.0)
104 staleness = gen_array_ops.reshape(
|
| /external/tensorflow/tensorflow/python/ops/ |
| array_ops.py | 102 from tensorflow.python.ops import gen_array_ops
106 from tensorflow.python.ops.gen_array_ops import *
132 return gen_array_ops.identity(input, name=name)
198 return gen_array_ops._expand_dims(input, axis, name)
211 return gen_array_ops._list_diff(x, y, out_idx, name)
214 listdiff.__doc__ = gen_array_ops._list_diff.__doc__ + "\n" + listdiff.__doc__
222 return gen_array_ops._list_diff(x, y, index_dtype, name)
225 setdiff1d.__doc__ = gen_array_ops._list_diff.__doc__
242 return gen_array_ops._broadcast_args(shape_x, shape_y)
314 return gen_array_ops.shape(input, name=name, out_type=out_type
[all...] |
| math_grad.py | 28 from tensorflow.python.ops import gen_array_ops
620 unused_ra, rx = gen_array_ops._broadcast_gradient_args(sa, sx)
649 _, rx = gen_array_ops._broadcast_gradient_args(sa, sx)
677 unused_rx, rq = gen_array_ops._broadcast_gradient_args(sx, sq)
699 unused_rn, rx = gen_array_ops._broadcast_gradient_args(sn, sx)
843 rx, ry = gen_array_ops._broadcast_gradient_args(sx, sy)
860 rx, ry = gen_array_ops._broadcast_gradient_args(sx, sy)
879 rx, ry = gen_array_ops._broadcast_gradient_args(sx, sy)
897 rx, ry = gen_array_ops._broadcast_gradient_args(sx, sy)
922 rx, ry = gen_array_ops._broadcast_gradient_args(sx, sy
[all...] |
| array_grad.py | 31 from tensorflow.python.ops import gen_array_ops
151 offset = gen_array_ops._concat_offset(non_neg_concat_dim, sizes)
631 return gen_array_ops._reverse(grad, reverse_dims), None
704 return [gen_array_ops._mirror_pad_grad(grad, op.inputs[1], mode=mode), None]
712 return [gen_array_ops._mirror_pad(grad, op.inputs[1], mode=mode), None]
|
| /external/tensorflow/tensorflow/python/eager/ |
| custom_gradient.py | 25 from tensorflow.python.ops import gen_array_ops
78 flat_result = [gen_array_ops.identity(x) for x in flat_result]
|
| benchmarks_test.py | 43 from tensorflow.python.ops import gen_array_ops
195 self._run(lambda: gen_array_ops.identity(m), 30000)
212 lambda: backprop.gradients_function(gen_array_ops.identity, [0])(m),
219 self._run(lambda: gen_array_ops.identity(m), 30000)
|
| backprop.py | 39 from tensorflow.python.ops import gen_array_ops
569 args[i] = gen_array_ops.identity(args[i])
696 flat_result = [gen_array_ops.identity(x) for x in flat_result]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| identity_op_py_test.py | 26 from tensorflow.python.ops import gen_array_ops
68 self.assertEquals(shape, gen_array_ops._ref_identity(tensor).get_shape())
|
| unique_op_test.py | 25 from tensorflow.python.ops import gen_array_ops
69 y0, idx0 = gen_array_ops._unique_v2(x, axis=np.array([0], dtype))
71 y1, idx1 = gen_array_ops._unique_v2(x, axis=np.array([1], dtype))
83 y, idx = gen_array_ops._unique_v2(x, axis=np.array([], np.int32))
|
| concat_op_test.py | 27 from tensorflow.python.ops import gen_array_ops
529 output = gen_array_ops._concat_v2([t1, t2], 0).eval()
537 gen_array_ops._concat_v2([t1, t2], 1).eval()
544 c = gen_array_ops._concat_v2([t1, t2], -2)
550 c = gen_array_ops._concat_v2([t1, t2], -1)
618 off = gen_array_ops._concat_offset(cdim, [s0, s1, s2])
627 off = gen_array_ops._concat_offset(cdim, [s0, s1])
637 off = gen_array_ops._concat_offset(cdim, [s0, s1])
647 off = gen_array_ops._concat_offset(cdim, [s0, s1])
657 off = gen_array_ops._concat_offset(cdim, [s0, s1]
[all...] |
| bcast_ops_test.py | 23 from tensorflow.python.ops.gen_array_ops import _broadcast_args
24 from tensorflow.python.ops.gen_array_ops import _broadcast_gradient_args
|
| depthtospace_op_test.py | 29 from tensorflow.python.ops import gen_array_ops
243 t, _, _ = gen_array_ops.quantize_v2(t, -128.0, 127.0, dtypes.qint8)
245 t = gen_array_ops.dequantize(t, -128, 127)
|
| spacetodepth_op_test.py | 28 from tensorflow.python.ops import gen_array_ops
241 t, _, _ = gen_array_ops.quantize_v2(t, -128.0, 127.0, dtypes.qint8)
243 t = gen_array_ops.dequantize(t, -128, 127)
|
| batchtospace_op_test.py | 31 from tensorflow.python.ops import gen_array_ops
47 return gen_array_ops._batch_to_space(*args, **kwargs)
|
| /external/tensorflow/tensorflow/contrib/nn/python/ops/ |
| scaled_softplus.py | 24 from tensorflow.python.ops import gen_array_ops
34 bcast_dims, _ = gen_array_ops._broadcast_gradient_args(shape, g_shape)
|
| /external/tensorflow/tensorflow/python/training/ |
| adagrad.py | 23 from tensorflow.python.ops import gen_array_ops
74 init_constant = gen_array_ops.fill(array_ops.shape(v),
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| spacetobatch_op_test.py | 25 from tensorflow.python.ops import gen_array_ops
78 x_tf = gen_array_ops._space_to_batch(
82 x_tf = gen_array_ops._batch_to_space(
|
| fake_quant_ops_test.py | 23 from tensorflow.python.ops import gen_array_ops
207 outputs = gen_array_ops.fake_quant_with_min_max_args_gradient(
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| debug_gradients.py | 29 from tensorflow.python.ops import gen_array_ops
159 identity_op = (gen_array_ops._debug_gradient_ref_identity
161 else gen_array_ops._debug_gradient_identity)
|
| /external/tensorflow/tensorflow/contrib/crf/python/ops/ |
| crf.py | 59 from tensorflow.python.ops import gen_array_ops
438 gen_array_ops.gather_nd(inputs, indices), # [B]
489 backpointers = gen_array_ops.reverse_sequence( # [B, T - 1, O]
507 decode_tags = gen_array_ops.reverse_sequence( # [B, T]
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| fused_conv2d_bias_activation_op_test.py | 29 from tensorflow.python.ops import gen_array_ops
643 NchwVectCToNchw(gen_array_ops.dequantize(conv_input, -128, 127)),
644 OihwVectIToHwio(gen_array_ops.dequantize(kernel, -128, 127)),
650 gen_array_ops.dequantize(side_input, -128, 127))
654 result, _, _ = gen_array_ops.quantize_v2(
813 conv_input, _, _ = gen_array_ops.quantize_v2(
820 kernel, _, _ = gen_array_ops.quantize_v2(
836 side_input, _, _ = gen_array_ops.quantize_v2(
[all...] |
| /external/tensorflow/tensorflow/python/grappler/ |
| item_test.py | 29 from tensorflow.python.ops import gen_array_ops
114 i = gen_array_ops._ref_identity(v)
|
