Searched defs:op_key (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/ |
| D | mutable_op_resolver.cc | 71 auto op_key = std::make_pair(op, version); in AddBuiltin() local
96 auto op_key = std::make_pair(name, version); in AddCustom() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| D | tfrt_fallback_common.cc | 45 mlir::IntegerAttr op_key; in ParseExecuteOpCommon() local
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | loader_impl.py | 201 def _get_op_from_collection(meta_graph_def, op_key): argument
|
| /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
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_execute_compat.cc | 638 tfrt::I64Attr op_key() const { in op_key() function in tensorflow::tfd::__anon6ff0a4040911::FallbackKernelAttributeFrame
739 tfrt::I64Attr op_key, tfrt::StringAttr op_name_attr, in KernelFallbackCreateOp()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 408 RngOutput PhiloxRngBit32(XlaOp op_key, XlaOp initial_state, in PhiloxRngBit32()
435 RngOutput PhiloxRngBit64(XlaOp op_key, XlaOp initial_state, in PhiloxRngBit64()
|