Searched refs:handle_op (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_hashtables.cc | 79 auto handle_op = find_op.table_handle().getDefiningOp(); in matchAndRewrite() local
80 if (handle_op == nullptr) return failure(); in matchAndRewrite()
81 auto hashtable_op = llvm::dyn_cast<TFL::HashtableOp>(handle_op); in matchAndRewrite()
97 auto handle_op = import_op.table_handle().getDefiningOp(); in matchAndRewrite() local
98 if (handle_op == nullptr) return failure(); in matchAndRewrite()
99 auto hashtable_op = llvm::dyn_cast<TFL::HashtableOp>(handle_op); in matchAndRewrite()
115 auto handle_op = size_op.table_handle().getDefiningOp(); in matchAndRewrite() local
116 if (handle_op == nullptr) return failure(); in matchAndRewrite()
117 auto hashtable_op = llvm::dyn_cast<TFL::HashtableOp>(handle_op); in matchAndRewrite()
|
| /external/mesa3d/src/compiler/glsl/ |
| D | lower_int64.cpp | 123 ir_rvalue *handle_op(ir_expression *ir, const char *function_name,
316 lower_64bit_visitor::handle_op(ir_expression *ir, in handle_op() function in lower_64bit_visitor
358 *rvalue = handle_op(ir, "__builtin_sign64", generate_ir::sign64); in handle_rvalue()
365 *rvalue = handle_op(ir, "__builtin_udiv64", generate_ir::udiv64); in handle_rvalue()
367 *rvalue = handle_op(ir, "__builtin_idiv64", generate_ir::idiv64); in handle_rvalue()
375 *rvalue = handle_op(ir, "__builtin_umod64", generate_ir::umod64); in handle_rvalue()
377 *rvalue = handle_op(ir, "__builtin_imod64", generate_ir::imod64); in handle_rvalue()
384 *rvalue = handle_op(ir, "__builtin_umul64", generate_ir::umul64); in handle_rvalue()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib_test.cc | 56 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op( in TEST() local
59 TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT); in TEST()
60 TFE_OpSetAttrShape(handle_op.get(), "shape", /*dims=*/nullptr, /*num_dims=*/0, in TEST()
65 "VarHandleOp", TFE_OpGetAttrs(handle_op.get()), in TEST()
86 "VarHandleOp", TFE_OpGetAttrs(handle_op.get()), in TEST()
112 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op( in TEST() local
115 TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT); in TEST()
116 TFE_OpSetAttrShape(handle_op.get(), "shape", /*dims=*/nullptr, /*num_dims=*/0, in TEST()
121 "VarHandleOp", TFE_OpGetAttrs(handle_op.get()), in TEST()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | variable_merger_pass.cc | 94 Status MergeReadVariableOps(Node* handle_op, Node* control_node, in MergeReadVariableOps() argument
109 node_def.set_device(handle_op->requested_device()); in MergeReadVariableOps()
111 node->set_assigned_device_name(handle_op->assigned_device_name()); in MergeReadVariableOps()
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | saveable_object_util.py | 94 self.handle_op = var.op.inputs[0]
114 self.handle_op = var.handle
134 self.handle_op, self._var_shape, restored_tensor)
|
| /external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| D | pass.cc | 904 Operation *handle_op = op->getOperand(0).getDefiningOp(); in matchAndRewrite() local
905 if (!handle_op || handle_op->getNumOperands() == 0) return failure(); in matchAndRewrite()
907 auto dynamic_size = handle_op->getAttrOfType<BoolAttr>("dynamic_size"); in matchAndRewrite()
910 Operation *array_size = handle_op->getOperand(0).getDefiningOp(); in matchAndRewrite()
921 control_operands.push_back(TFOp(handle_op).controlRet()); in matchAndRewrite()
|