Searched refs:cost_estimate_ns (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| D | hlo_parser.cc | 999 optional<int64> cost_estimate_ns; in ParseInstruction() local
1003 &cost_estimate_ns}; in ParseInstruction()
1008 shape, operands, *channel_name, *cost_estimate_ns)); in ParseInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | computation_builder.cc | 794 const string& channel_name, int64 cost_estimate_ns, const Shape& shape) { in HostCompute() argument
802 request->set_cost_estimate_ns(cost_estimate_ns); in HostCompute()
|
| D | computation_builder.h | 457 const string& channel_name, int64 cost_estimate_ns, const Shape& shape);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.h | 488 tensorflow::StringPiece channel_name, const int64 cost_estimate_ns);
|
| D | user_computation.cc | 3395 auto cost_estimate_ns = host_compute_request.cost_estimate_ns(); in Visit() local
3397 output_shape, operands, channel_name, cost_estimate_ns)); in Visit()
|
| D | hlo_instruction.cc | 1137 tensorflow::StringPiece channel_name, const int64 cost_estimate_ns) { in CreateHostCompute() argument
1144 instruction->cost_estimate_ns_ = cost_estimate_ns; in CreateHostCompute()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 557 int64 cost_estimate_ns = 3; field
|