| /external/tensorflow/tensorflow/core/kernels/ |
| D | adjust_hsv_gpu.cu.h | 75 const float new_h = h * 6.0f; in hsv2rgb_cuda() local
77 const float x = chroma * (1.0f - fabsf(fmodf(new_h, 2.0f) - 1.0f)); in hsv2rgb_cuda()
79 const bool between_0_and_1 = new_h >= 0.0f && new_h < 1.0f; in hsv2rgb_cuda()
80 const bool between_1_and_2 = new_h >= 1.0f && new_h < 2.0f; in hsv2rgb_cuda()
81 const bool between_2_and_3 = new_h >= 2.0f && new_h < 3.0f; in hsv2rgb_cuda()
82 const bool between_3_and_4 = new_h >= 3.0f && new_h < 4.0f; in hsv2rgb_cuda()
83 const bool between_4_and_5 = new_h >= 4.0f && new_h < 5.0f; in hsv2rgb_cuda()
84 const bool between_5_and_6 = new_h >= 5.0f && new_h < 6.0f; in hsv2rgb_cuda()
116 float new_h = hsv.h; in adjust_hsv_nhwc() local
122 new_h = fmodf(hsv.h + delta, 1.0f); in adjust_hsv_nhwc()
[all …]
|
| /external/compiler-rt/lib/tsan/rtl/ |
| D | tsan_libdispatch_mac.cc | 514 fd_handler_t new_h = Block_copy(^(dispatch_data_t data, int error) { in TSAN_INTERCEPTOR() local
522 REAL(dispatch_read)(fd, length, q, new_h); in TSAN_INTERCEPTOR()
523 Block_release(new_h); in TSAN_INTERCEPTOR()
531 fd_handler_t new_h = Block_copy(^(dispatch_data_t data, int error) { in TSAN_INTERCEPTOR() local
539 REAL(dispatch_write)(fd, data, q, new_h); in TSAN_INTERCEPTOR()
540 Block_release(new_h); in TSAN_INTERCEPTOR()
548 dispatch_io_handler_t new_h = in TSAN_INTERCEPTOR() local
557 REAL(dispatch_io_read)(channel, offset, length, q, new_h); in TSAN_INTERCEPTOR()
558 Block_release(new_h); in TSAN_INTERCEPTOR()
567 dispatch_io_handler_t new_h = in TSAN_INTERCEPTOR() local
[all …]
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent_v2.py | 279 last_output, outputs, new_h, runtime = cudnn_gru(
287 last_output, outputs, new_h, runtime = standard_gru(
304 last_output, outputs, new_h, runtime = defun_standard_gru(
317 states = [new_h]
632 last_output, outputs, new_h, new_c, runtime = cudnn_lstm(
636 last_output, outputs, new_h, new_c, runtime = standard_lstm(
653 last_output, outputs, new_h, new_c, runtime = defun_standard_lstm(
662 states = [new_h, new_c]
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | gru_ops.py | 206 _, _, _, new_h = _gru_block_cell(
214 return new_h, new_h
|
| D | rnn_cell.py | 1460 new_h = self._activation(new_c) * math_ops.sigmoid(o)
1462 new_state = rnn_cell_impl.LSTMStateTuple(new_c, new_h)
1463 return new_h, new_state
2033 new_h = math_ops.tanh(new_c) * output_gate
2058 new_h = k * new_h + (1 - k) * h_prev
2060 new_state = rnn_cell_impl.LSTMStateTuple(new_c, new_h)
2062 return new_h, new_state
3062 new_h = sigmoid(o + w_o_diag * new_c) * self._activation(new_c)
3064 new_h = sigmoid(o) * self._activation(new_c)
3068 new_h = self._linear(
[all …]
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| D | rnn_cells.py | 163 new_h = multiply(self._activation(new_c), sigmoid(o))
166 new_state = tf_rnn.LSTMStateTuple(new_c, new_h)
168 new_state = array_ops.concat([new_c, new_h], 1)
169 return new_h, new_state
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_cell_impl.py | 594 new_h = u * state + (1 - u) * c
595 return new_h, new_h
781 new_h = multiply(self._activation(new_c), sigmoid(o))
784 new_state = LSTMStateTuple(new_c, new_h)
786 new_state = array_ops.concat([new_c, new_h], 1)
787 return new_h, new_state
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner_test.cc | 182 const NodeDef& new_h = output.node(6); in TEST_F() local
183 EXPECT_EQ("h", new_h.name()); in TEST_F()
195 EXPECT_EQ(1, new_h.input_size()); in TEST_F()
196 EXPECT_EQ("f:1", new_h.input(0)); in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 176 new_h = (1-u) * candidate + u * state
177 return new_h, new_h
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | rnn_cell_test.py | 1020 new_c, new_h = res[1]
1023 self.assertEqual(new_h.shape[0], batch_size)
1024 self.assertEqual(new_h.shape[1], num_units)
1071 new_c, new_h = res[1]
1074 self.assertEqual(new_h.shape[0], batch_size)
1075 self.assertEqual(new_h.shape[1], num_proj)
|