| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/analysis/ |
| H A D | test_cost_analysis_pass.cc | 35 int64_t op_key = 0; in runOnOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/ |
| H A D | cost_recorder.cc | 30 void CostRecorder::RecordCost(int64_t op_key, const uint64_t execution_time) { in RecordCost()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| H A D | tfrt_fallback_common.cc | 45 mlir::IntegerAttr op_key; in ParseExecuteOpCommon() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/ |
| H A D | tfrt_fallback_common.cc | 45 mlir::IntegerAttr op_key; in ParseExecuteOpCommon() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | kernel_fallback_execute_compat.cc | 462 tfrt::I64Attr op_key() const { in op_key() function in tensorflow::tfd::__anond671884e0711::FallbackKernelAttributeFrame
603 tfrt::I64Attr op_key, tfrt::StringAttr op_name_attr, in KernelFallbackCreateOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/ |
| H A D | loader_impl.py | 193 def _get_op_from_collection(meta_graph_def, op_key): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/logging/ |
| H A D | conversion_log_util.cc | 160 tflite::details::OperatorKey op_key(op_signature, ops_by_type, in CountOperatorsByType() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | handle_cross_cluster_dependencies.cc | 192 const std::string op_key = in LowerToSendRecv() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/ |
| H A D | xplane_to_op_metrics_db.cc | 165 OpKey op_key; in GetOpKeyFromHloEventMetadata() local
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | __init__.py | 2509 def registerOp(cls, op_key, full_schema, op_impl, dispatch_key): argument
|