Searched refs:op_input (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | argminmax_test.py | 32 def _assertOpOutputMatchesExpected(self, op, axis, output_type, op_input, argument
46 dtypes.as_dtype(op_input.dtype), op_input.shape, name="a")
48 result = session.run(output, {pinp: op_input})
63 op_input=np.array([1, 10, 27, 3, 3, 4], dtype=dtype),
69 op_input=np.array([[4, 1, 7], [3, 2, 4]], dtype=dtype),
75 op_input=np.array([[4, 1], [3, 2]], dtype=dtype),
82 op_input=np.array([3, 10, 27, 3, 2, 4], dtype=dtype),
88 op_input=np.array([[4, 1, 7], [3, 2, 4]], dtype=dtype),
94 op_input=np.array([[4, 1], [3, 2]], dtype=dtype),
|
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | input_to_ops.py | 45 for op_input in op.inputs:
46 self.mapping[op_input].add(op)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 280 for i, op_input in enumerate(op_inputs):
281 if uses_keras_history(op_input):
282 layer_inputs.append(op_input)
288 constants[i] = backend.function([], op_input)([])
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variables.py | 2354 for op_input in op.inputs:
2355 if _has_cycle(op_input.op, path):
2414 for op_input in op.inputs:
2415 new_op_input = _safe_initial_value_from_tensor(name, op_input, op_cache)
2417 modified = modified or (new_op_input != op_input)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 1800 for op_input in inputs:
1801 if isinstance(op_input, (list, tuple)):
1802 c_api.TF_AddInputList(op_desc, [t._as_tf_output() for t in op_input])
1804 c_api.TF_AddInput(op_desc, op_input._as_tf_output())
5951 for op_input in op_input_list:
5956 if (isinstance(op_input, (Operation, _TensorLike)) and
5957 …((not isinstance(op_input, Tensor)) or type(op_input) == Tensor)): # pylint: disable=unidiomatic-…
5958 graph_element = op_input
5960 graph_element = _as_graph_element(op_input)
|
| D | ops_test.py | 807 op_input = op.inputs[0].op
808 self.assertEqual(op_input.type, "Switch")
809 self.assertEqual(op_input.inputs[0], x)
837 op_input = op.inputs[0].op
838 self.assertEqual(op_input.type, "Enter")
839 self.assertEqual(list(op_input.inputs), [x])
|