Home
last modified time | relevance | path

Searched refs:op_func (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dmath_test.py519 def test_sparse_segment(self, op_func, with_num_segments): argument
536 op_func(data_0, indices_i, seg_ids_0, num_segments=num_segments),
537 op_func(data_i, indices_i, seg_ids_0, num_segments=num_segments),
538 op_func(data_0, indices_0, seg_ids_0, num_segments=num_segments),
539 op_func(data_i, indices_0, seg_ids_0, num_segments=num_segments)
544 op_func(data_0, indices_i, seg_ids_i, num_segments=num_segments),
545 op_func(data_i, indices_i, seg_ids_i, num_segments=num_segments),
546 op_func(data_0, indices_0, seg_ids_i, num_segments=num_segments),
547 op_func(data_i, indices_0, seg_ids_i, num_segments=num_segments)
555 def test_sparse_segment_grad(self, op_func): argument
[all …]
Dpfor.py2045 def _convert_softmax(pfor_input, op_type, op_func): argument
2047 return wrap(op_func(pfor_input.stacked_input(0)), True)
2057 def _convert_identity(pfor_input, op_type, op_func): argument
2059 return wrap(op_func(*[x.t for x in pfor_input.inputs]), True)
2123 def _convert_searchsorted(pfor_input, _, op_func): argument
2128 output = op_func(sorted_inputs, values, out_type)
2321 def _convert_transpose(pfor_input, _, op_func): argument
2325 return wrap(op_func(t, new_perm), True)
2599 def _convert_reduction(pfor_input, _, op_func): argument
2605 return wrap(op_func(t, indices, keepdims=keep_dims), True)
[all …]
Dcontrol_flow_ops_test.py2348 def test_fft(self, op_func): argument
2354 return op_func(x_i)
2363 def test_rfft(self, op_func): argument
2370 return op_func(x_i)
2381 def test_irfft(self, op_func): argument
2393 return op_func(x_i)
/external/deqp/modules/gles31/scripts/
Dgen-implicit-conversions.py341 self.op_func = {
362 assert(self.op in self.op_func)
368 return self.op_func[self.op](a, b)
409 self.op_func = {
419 assert(self.op in self.op_func)
427 return Scalar(self.op_func[self.op](a, b))
/external/autotest/client/common_lib/hosts/
Dbase_classes.py552 def log_op(self, op, op_func): argument
563 self.job.run_op(op, op_func, self.get_kernel_ver)
567 op_func()
/external/autotest/server/hosts/
Dremote.py246 def op_func(): function
249 self.log_op(self.OP_REBOOT, op_func)
/external/tensorflow/tensorflow/python/ops/
Dcheck_ops.py301 def _binary_assert(sym, opname, op_func, static_func, x, y, data, summarize, argument
333 test_op = op_func(x, y)
367 condition = math_ops.reduce_all(op_func(x, y))
/external/tensorflow/tensorflow/python/framework/
Dpython_api_parameter_converter_test.py58 def makeApiInfoForGenOp(self, op_name, op_func): argument
/external/autotest/server/
Dserver_job.py1056 def run_op(self, op, op_func, get_kernel_func): argument
1070 op_func()