Searched refs:tmp_var (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | variable_ops.cc | 111 auto* tmp_var = new TmpVar; in Compute() local 112 OP_REQUIRES(context, tmp_var, in Compute() 114 tmp_var->name = unique_name; in Compute() 115 s = context->allocate_temp(dtype_, shape_, &tmp_var->val); in Compute() 116 if (!s.ok()) tmp_var->Unref(); in Compute() 119 context->step_container()->Create(rm, unique_name, tmp_var)); in Compute() 120 context->set_output_ref(0, &tmp_var->mu, &tmp_var->val); in Compute() 123 tmp_var->val.AllocatedBytes()); in Compute()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | memory_optimizer.cc | 673 NodeDef* tmp_var = item->graph.add_node(); in SchedulingPass() local 674 tmp_var->set_name(tmp_var_name); in SchedulingPass() 675 tmp_var->set_op("TemporaryVariable"); in SchedulingPass() 676 tmp_var->set_device(device); in SchedulingPass() 677 (*tmp_var->mutable_attr())["dtype"].set_type(dtype); in SchedulingPass() 678 *(*tmp_var->mutable_attr())["shape"].mutable_shape() = shape; in SchedulingPass() 679 (*tmp_var->mutable_attr())["var_name"].set_s(tmp_var->name()); in SchedulingPass() 682 *tmp_var->add_input() = ctrl_dep; in SchedulingPass() 684 *tmp_var->add_input() = in SchedulingPass() 702 *initialize->add_input() = tmp_var->name(); in SchedulingPass() [all …]
|
/external/llvm-project/compiler-rt/cmake/Modules/ |
D | CompilerRTDarwinUtils.cmake | 176 list_intersect(tmp_var DARWIN_osx_ARCHS ${input}) 191 list(REMOVE_ITEM tmp_var i386) 193 list(REMOVE_ITEM tmp_var arm64) 194 list(REMOVE_ITEM tmp_var arm64e) 208 list(REMOVE_ITEM tmp_var x86_64h) 212 set(${output} ${tmp_var} PARENT_SCOPE)
|
/external/bcc/src/lua/bpf/ |
D | bpf.lua | 902 local tmp_var = stackslots + 1 903 vset(tmp_var, nil, d) 904 ALU_REG(tmp_var, tmp_var, b, 'ADD') 905 local dst_reg = vreg(tmp_var) 906 V[tmp_var].reg = nil -- Only temporary allocation 916 local tmp_var = stackslots + 1 917 vcopy(tmp_var, d) -- Element position 919 ALU_IMM(tmp_var, tmp_var, w, 'MUL') -- multiply by element size 921 local dst_reg = vreg(tmp_var) -- add R10 (stack pointer) 923 V[tmp_var].reg = nil -- Only temporary allocation
|
/external/compiler-rt/cmake/Modules/ |
D | CompilerRTDarwinUtils.cmake | 104 list_intersect(tmp_var DARWIN_osx_ARCHS ${input}) 118 list(REMOVE_ITEM tmp_var x86_64h) 120 set(${output} ${tmp_var} PARENT_SCOPE)
|