Searched refs:op_key (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/ |
| D | mutable_op_resolver.cc | 46 auto op_key = std::make_pair(op, version); in AddBuiltin() local
47 builtins_[op_key] = new_registration; in AddBuiltin()
65 auto op_key = std::make_pair(name, version); in AddCustom() local
66 custom_ops_[op_key] = new_registration; in AddCustom()
|
| /external/tensorflow/tensorflow/lite/toco/logging/ |
| D | conversion_log_util.cc | 157 tflite::details::OperatorKey op_key(op_signature, ops_by_type, in CountOperatorsByType() local
161 if (op_key.is_custom_op()) { in CountOperatorsByType()
163 } else if (op_key.is_flex_op()) { in CountOperatorsByType()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 407 RngOutput PhiloxRngBit32(XlaOp op_key, XlaOp initial_state, in PhiloxRngBit32() argument
409 XlaBuilder* builder = op_key.builder(); in PhiloxRngBit32()
412 Philox4x32Key key = Uint64ToUint32s(op_key); in PhiloxRngBit32()
434 RngOutput PhiloxRngBit64(XlaOp op_key, XlaOp initial_state, in PhiloxRngBit64() argument
436 XlaBuilder* builder = op_key.builder(); in PhiloxRngBit64()
439 Philox4x32Key key = Uint64ToUint32s(op_key); in PhiloxRngBit64()
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | loader_impl.py | 194 def _get_op_from_collection(meta_graph_def, op_key): argument
195 return _get_main_op_tensor(meta_graph_def, op_key)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 1454 op_key = (pooling_type, num_spatial_dims)
1455 if op_key not in pooling_ops:
1456 raise ValueError("%d-D %s pooling is not supported." % (op_key[1],
1457 op_key[0]))
1484 result = pooling_ops[op_key](
|