| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | cost_recorder.cc | 30 void CostRecorder::RecordCost(int64_t op_key, const uint64_t execution_time) { in RecordCost() argument
32 op_cost_map_[op_key].first += execution_time; in RecordCost()
33 op_cost_map_[op_key].second += 1; in RecordCost()
45 for (const auto& [op_key, op_cost] : op_cost_map_) { in WriteToFile()
47 (*op_cost_map_proto.mutable_op_cost_map())[op_key] = avg_op_cost; in WriteToFile()
|
| D | op_cost_map.proto | 9 // Maps an op_key to a cost measured in microseconds.
|
| D | cost_recorder.h | 41 void RecordCost(int64_t op_key, const uint64_t execution_time);
|
| /external/tensorflow/tensorflow/lite/ |
| D | mutable_op_resolver.cc | 71 auto op_key = std::make_pair(op, version); in AddBuiltin() local
72 builtins_[op_key] = new_registration; in AddBuiltin()
96 auto op_key = std::make_pair(name, version); in AddCustom() local
97 custom_ops_[op_key] = new_registration; in AddCustom()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/analysis/ |
| D | test_cost_analysis_pass.cc | 35 int64_t op_key = 0; in runOnOperation() local
37 op.emitRemark() << "Cost: " << cost_analysis.GetCost(&op, op_key++); in runOnOperation()
|
| D | cost_analysis.cc | 131 int64_t CostAnalysis::GetCost(mlir::Operation* op, int64_t op_key) const { in GetCost()
134 if (const auto op_cost = measured_cost_map.find(op_key); in GetCost()
|
| D | cost_analysis.h | 45 int64_t GetCost(mlir::Operation* op, int64_t op_key) const;
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | xplane_to_op_metrics_db.cc | 165 OpKey op_key; in GetOpKeyFromHloEventMetadata() local
170 op_key.program_id = stat.IntOrUintValue(); in GetOpKeyFromHloEventMetadata()
173 op_key.symbol_id = stat.IntOrUintValue(); in GetOpKeyFromHloEventMetadata()
180 return op_key; in GetOpKeyFromHloEventMetadata()
|
| /external/tensorflow/tensorflow/lite/toco/logging/ |
| D | conversion_log_util.cc | 160 tflite::details::OperatorKey op_key(op_signature, ops_by_type, in CountOperatorsByType() local
164 if (op_key.is_custom_op()) { in CountOperatorsByType()
166 } else if (op_key.is_flex_op()) { in CountOperatorsByType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/ |
| D | tfrt_fallback_async.td | 299 op_key: The unique key to identify this op.
319 I64Attr:$op_key,
364 I64Attr:$op_key,
382 "int64_t":$op_key, "::llvm::StringRef":$op_name, "int64_t":$_tfrt_cost)>];
427 I64Attr:$op_key,
493 I64Attr:$op_key,
526 I64Attr:$op_key,
|
| D | tfrt_fallback_sync.td | 92 op_key: The unique key to identify this op.
106 I64Attr:$op_key,
128 I64Attr:$op_key,
|
| D | tfrt_fallback_common.cc | 45 mlir::IntegerAttr op_key; in ParseExecuteOpCommon() local
63 parser.parseAttribute(op_key, "op_key", result.attributes) || in ParseExecuteOpCommon()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| D | tfrt_fallback_common.cc | 45 mlir::IntegerAttr op_key; in ParseExecuteOpCommon() local
63 parser.parseAttribute(op_key, "op_key", result.attributes) || in ParseExecuteOpCommon()
|
| D | tfrt_fallback_async.td | 297 op_key: The unique key to identify this op.
317 I64Attr:$op_key,
363 I64Attr:$op_key,
381 "int64_t":$op_key, "::llvm::StringRef":$op_name, "int64_t":$_tfrt_cost)>];
427 I64Attr:$op_key,
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | handle_cross_cluster_dependencies.cc | 192 const std::string op_key = in LowerToSendRecv() local
205 copy_to_mesh.getLoc(), value_to_send, builder.getStringAttr(op_key), in LowerToSendRecv()
218 builder.getStringAttr(op_key), in LowerToSendRecv()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 408 RngOutput PhiloxRngBit32(XlaOp op_key, XlaOp initial_state, in PhiloxRngBit32() argument
410 XlaBuilder* builder = op_key.builder(); in PhiloxRngBit32()
413 Philox4x32Key key = Uint64ToUint32s(op_key); in PhiloxRngBit32()
435 RngOutput PhiloxRngBit64(XlaOp op_key, XlaOp initial_state, in PhiloxRngBit64() argument
437 XlaBuilder* builder = op_key.builder(); in PhiloxRngBit64()
440 Philox4x32Key key = Uint64ToUint32s(op_key); in PhiloxRngBit64()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_execute_compat.cc | 462 tfrt::I64Attr op_key() const { in op_key() function in tensorflow::tfd::__anone1369dfd0711::FallbackKernelAttributeFrame
559 .RecordCost(frame.op_key().GetValue(), execution_time); in KernelFallbackExecuteOpInternal()
585 auto* kernel_runner = runner_table->Get(frame.op_key().GetValue()); in KernelFallbackExecuteOp()
603 tfrt::I64Attr op_key, tfrt::StringAttr op_name_attr, in KernelFallbackCreateOp() argument
633 if (!runner_table->Insert(op_key.GetValue(), in KernelFallbackCreateOp()
848 auto* kernel_runner = runner_table->Get(attr_frame.op_key().GetValue()); in KernelFallbackExecuteOpCustomAllocatorInternal()
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | loader_impl.py | 193 def _get_op_from_collection(meta_graph_def, op_key): argument
194 return _get_main_op_tensor(meta_graph_def, op_key)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 1632 op_key = (pooling_type, num_spatial_dims)
1633 if op_key not in pooling_ops:
1663 result = pooling_ops[op_key](
|