Searched refs:op_state (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 601 OperationState& op_state, Location loc, OpBuilder& builder) { in AddOpIntermediatesForLstm() argument
621 op_state.addAttribute(named_attr.first, named_attr.second); in AddOpIntermediatesForLstm()
648 OperationState op_state(loc, op_name); in ConvertOp() local
653 op_state.addOperands({optional_arg_marker}); in ConvertOp()
655 op_state.addOperands({vals_map.at(input_num)}); in ConvertOp()
670 op_state.addAttribute("qtype", mlir::TypeAttr::get(type)); in ConvertOp()
671 } else if (op_name == "tfl.reshape" && op_state.operands.size() == 1) { in ConvertOp()
685 op_state.addOperands({shape_op}); in ConvertOp()
688 op_state.addTypes({type}); in ConvertOp()
698 int op_input_num = op_state.operands.size(); in ConvertOp()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | tfr_decompose_ctx.cc | 163 mlir::OperationState op_state(loc, tf_op_full_name); in ExpandNode() local
164 op_state.addOperands(func.getArguments()); in ExpandNode()
165 op_state.addTypes(output_tys); in ExpandNode()
166 op_state.addAttributes(attrs); in ExpandNode()
167 mlir::Operation* tf_op = op_builder.createOperation(op_state); in ExpandNode()
|
| /external/rust/crates/grpcio-sys/grpc/src/core/ext/transport/cronet/transport/ |
| D | cronet_transport.cc | 153 struct op_state { struct
154 explicit op_state(grpc_core::Arena* arena) : rs(arena) {} in op_state() function
181 struct op_state state; argument
209 struct op_state state;
824 struct stream_obj* s, struct op_state* op_state, in op_can_be_run() argument
826 struct op_state* stream_state = &s->state; in op_can_be_run()
857 if (op_id == OP_RECV_MESSAGE && op_state->state_op_done[OP_RECV_MESSAGE]) { in op_can_be_run()
897 if (op_state->state_op_done[OP_SEND_MESSAGE]) { in op_can_be_run()
907 if (op_state->state_op_done[OP_RECV_MESSAGE]) { in op_can_be_run()
955 if (op_state->state_op_done[OP_ON_COMPLETE]) { in op_can_be_run()
[all …]
|
| /external/grpc-grpc/src/core/ext/transport/cronet/transport/ |
| D | cronet_transport.cc | 142 struct op_state { struct
165 struct op_state state; argument
187 struct op_state state;
806 struct stream_obj* s, struct op_state* op_state, in op_can_be_run() argument
808 struct op_state* stream_state = &s->state; in op_can_be_run()
839 if (op_id == OP_RECV_MESSAGE && op_state->state_op_done[OP_RECV_MESSAGE]) { in op_can_be_run()
877 if (op_state->state_op_done[OP_SEND_MESSAGE]) result = false; in op_can_be_run()
883 if (op_state->state_op_done[OP_RECV_MESSAGE]) result = false; in op_can_be_run()
923 if (op_state->state_op_done[OP_ON_COMPLETE]) { in op_can_be_run()
937 !op_state->state_op_done[OP_SEND_MESSAGE]) { in op_can_be_run()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variables.py | 2775 op_state = state.get(op.name, _UNKNOWN)
2776 if op_state == _STARTED:
2778 elif op_state == _FINISHED:
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 707 mlir::OperationState op_state(loc, mhlo_op_name); in EmitUsingElementalIrEmitter() local
711 mlir::Region* new_region = op_state.addRegion(); in EmitUsingElementalIrEmitter()
715 op_state.addOperands(loads); in EmitUsingElementalIrEmitter()
716 op_state.addAttributes(input.op->getAttrs()); in EmitUsingElementalIrEmitter()
717 op_state.addTypes({mlir::RankedTensorType::get( in EmitUsingElementalIrEmitter()
720 new_op = b.createOperation(op_state); in EmitUsingElementalIrEmitter()
|